change exchange to ai_exchange

README.md

@@ -20,8 +20,8 @@
 > `export OPENAI_API_KEY=your-key-here`
 
 ``` python
-from exchange import Exchange, Message, Tool
-from exchange.providers import OpenAiProvider
+from ai_exchange import Exchange, Message, Tool
+from ai_exchange.providers import OpenAiProvider
 
 def word_count(text: str):
     """Get the count of words in text
@@ -85,11 +85,11 @@ Your plugin will then be available in your application or other applications bui
 through:
 
 ``` python
-from exchange.providers import get_provider
+from ai_exchange.providers import get_provider
 
 provider = get_provider('example').from_env()
 ```
 
 [CONTRIBUTING]: CONTRIBUTING.md
-[openaiprovider]: src/exchange/providers/openai.py
+[openaiprovider]: src/ai_exchange/providers/openai.py
 [plugins]: https://packaging.python.org/en/latest/guides/creating-and-discovering-plugins/

src/ai_exchange/__init__.py

@@ -0,0 +1,6 @@
+"""Classes for interacting with the exchange API."""
+
+from ai_exchange.tool import Tool  # noqa
+from ai_exchange.content import Text, ToolResult, ToolUse  # noqa
+from ai_exchange.message import Message  # noqa
+from ai_exchange.exchange import Exchange  # noqa

src/exchange/exchange.py → src/ai_exchange/exchange.py

@@ -6,12 +6,12 @@
 from attrs import define, evolve, field
 from tiktoken import get_encoding
 
-from exchange.checkpoint import Checkpoint
-from exchange.content import ToolResult, ToolUse
-from exchange.message import Message
-from exchange.moderators import ContextTruncate, Moderator
-from exchange.providers import Provider, Usage
-from exchange.tool import Tool
+from ai_exchange.checkpoint import Checkpoint
+from ai_exchange.content import ToolResult, ToolUse
+from ai_exchange.message import Message
+from ai_exchange.moderators import ContextTruncate, Moderator
+from ai_exchange.providers import Provider, Usage
+from ai_exchange.tool import Tool
 
 
 def validate_tool_output(output: str) -> None:

src/exchange/message.py → src/ai_exchange/message.py

@@ -6,8 +6,8 @@
 from attrs import define, field
 from jinja2 import Environment, FileSystemLoader
 
-from exchange.content import CONTENT_TYPES, Content, Text, ToolResult, ToolUse
-from exchange.utils import create_object_id
+from ai_exchange.content import CONTENT_TYPES, Content, Text, ToolResult, ToolUse
+from ai_exchange.utils import create_object_id
 
 Role = Literal["user", "assistant"]
 

src/ai_exchange/moderators/__init__.py

@@ -0,0 +1,13 @@
+from functools import cache
+from typing import Type
+
+from ai_exchange.moderators.base import Moderator
+from ai_exchange.utils import load_plugins
+from ai_exchange.moderators.passive import PassiveModerator  # noqa
+from ai_exchange.moderators.truncate import ContextTruncate  # noqa
+from ai_exchange.moderators.summarizer import ContextSummarizer  # noqa
+
+
+@cache
+def get_moderator(name: str) -> Type[Moderator]:
+    return load_plugins(group="exchange.moderator")[name]

src/exchange/moderators/passive.py → src/ai_exchange/moderators/passive.py

@@ -1,5 +1,5 @@
 from typing import Type
-from exchange.moderators.base import Moderator
+from ai_exchange.moderators.base import Moderator
 
 
 class PassiveModerator(Moderator):

src/exchange/moderators/summarizer.py → src/ai_exchange/moderators/summarizer.py

@@ -1,8 +1,8 @@
 from typing import List, Literal, Optional, Tuple, Type, Union
 
-from exchange import Message
-from exchange.checkpoint import Checkpoint
-from exchange.moderators import ContextTruncate, Moderator, PassiveModerator
+from ai_exchange import Message
+from ai_exchange.checkpoint import Checkpoint
+from ai_exchange.moderators import ContextTruncate, Moderator, PassiveModerator
 
 MAX_TOKENS = 112000
 SUMMARIZATION_OFFSET = 40000  # Keep a max of this many tokens

src/exchange/moderators/truncate.py → src/ai_exchange/moderators/truncate.py

@@ -1,7 +1,7 @@
 from typing import Literal, Optional, Type
 
-from exchange.moderators import PassiveModerator
-from exchange.moderators.base import Moderator
+from ai_exchange.moderators import PassiveModerator
+from ai_exchange.moderators.base import Moderator
 
 MAX_TOKENS = 112000
 

src/ai_exchange/providers/__init__.py

@@ -0,0 +1,14 @@
+from functools import cache
+from typing import Type
+
+from ai_exchange.providers.anthropic import AnthropicProvider  # noqa
+from ai_exchange.providers.base import Provider, Usage  # noqa
+from ai_exchange.providers.databricks import DatabricksProvider  # noqa
+from ai_exchange.providers.openai import OpenAiProvider  # noqa
+from ai_exchange.providers.azure import AzureProvider  # noqa
+from ai_exchange.utils import load_plugins
+
+
+@cache
+def get_provider(name: str) -> Type[Provider]:
+    return load_plugins(group="exchange.provider")[name]

src/exchange/providers/anthropic.py → src/ai_exchange/providers/anthropic.py

@@ -4,10 +4,10 @@
 
 import httpx
 
-from exchange import Message, Tool
-from exchange.content import Text, ToolResult, ToolUse
-from exchange.providers.base import Provider, Usage
-from exchange.providers.utils import raise_for_status
+from ai_exchange import Message, Tool
+from ai_exchange.content import Text, ToolResult, ToolUse
+from ai_exchange.providers.base import Provider, Usage
+from ai_exchange.providers.utils import raise_for_status
 
 ANTHROPIC_HOST = "https://api.anthropic.com/v1/messages"
 

src/exchange/providers/azure.py → src/ai_exchange/providers/azure.py

@@ -3,16 +3,16 @@
 
 import httpx
 
-from exchange.message import Message
-from exchange.providers.base import Provider, Usage
-from exchange.providers.utils import (
+from ai_exchange.message import Message
+from ai_exchange.providers.base import Provider, Usage
+from ai_exchange.providers.utils import (
     messages_to_openai_spec,
     openai_response_to_message,
     openai_single_message_context_length_exceeded,
     raise_for_status,
     tools_to_openai_spec,
 )
-from exchange.tool import Tool
+from ai_exchange.tool import Tool
 
 
 class AzureProvider(Provider):

src/exchange/providers/base.py → src/ai_exchange/providers/base.py

@@ -2,8 +2,8 @@
 from attrs import define, field
 from typing import List, Tuple, Type
 
-from exchange.message import Message
-from exchange.tool import Tool
+from ai_exchange.message import Message
+from ai_exchange.tool import Tool
 
 
 @define

src/exchange/providers/bedrock.py → src/ai_exchange/providers/bedrock.py

@@ -9,11 +9,11 @@
 
 import httpx
 
-from exchange.content import Text, ToolResult, ToolUse
-from exchange.message import Message
-from exchange.providers import Provider, Usage
-from exchange.providers.utils import raise_for_status
-from exchange.tool import Tool
+from ai_exchange.content import Text, ToolResult, ToolUse
+from ai_exchange.message import Message
+from ai_exchange.providers import Provider, Usage
+from ai_exchange.providers.utils import raise_for_status
+from ai_exchange.tool import Tool
 
 SERVICE = "bedrock-runtime"
 UTC = timezone.utc

src/exchange/providers/databricks.py → src/ai_exchange/providers/databricks.py

@@ -3,15 +3,15 @@
 
 import httpx
 
-from exchange.message import Message
-from exchange.providers.base import Provider, Usage
-from exchange.providers.utils import (
+from ai_exchange.message import Message
+from ai_exchange.providers.base import Provider, Usage
+from ai_exchange.providers.utils import (
     messages_to_openai_spec,
     openai_response_to_message,
     raise_for_status,
     tools_to_openai_spec,
 )
-from exchange.tool import Tool
+from ai_exchange.tool import Tool
 
 
 class DatabricksProvider(Provider):

src/exchange/providers/openai.py → src/ai_exchange/providers/openai.py

@@ -3,16 +3,16 @@
 
 import httpx
 
-from exchange.message import Message
-from exchange.providers.base import Provider, Usage
-from exchange.providers.utils import (
+from ai_exchange.message import Message
+from ai_exchange.providers.base import Provider, Usage
+from ai_exchange.providers.utils import (
     messages_to_openai_spec,
     openai_response_to_message,
     openai_single_message_context_length_exceeded,
     raise_for_status,
     tools_to_openai_spec,
 )
-from exchange.tool import Tool
+from ai_exchange.tool import Tool
 
 OPENAI_HOST = "https://api.openai.com/"
 

src/exchange/providers/utils.py → src/ai_exchange/providers/utils.py

@@ -4,9 +4,9 @@
 from typing import Any, Dict, List, Tuple
 
 import httpx
-from exchange.content import Text, ToolResult, ToolUse
-from exchange.message import Message
-from exchange.tool import Tool
+from ai_exchange.content import Text, ToolResult, ToolUse
+from ai_exchange.message import Message
+from ai_exchange.tool import Tool
 
 
 def raise_for_status(response: httpx.Response) -> httpx.Response:

src/exchange/tool.py → src/ai_exchange/tool.py

@@ -3,7 +3,7 @@
 
 from attrs import define
 
-from exchange.utils import json_schema, parse_docstring
+from ai_exchange.utils import json_schema, parse_docstring
 
 
 @define

tests/providers/test_anthropic.py

@@ -3,10 +3,10 @@
 
 import httpx
 import pytest
-from exchange import Message, Text
-from exchange.content import ToolResult, ToolUse
-from exchange.providers.anthropic import AnthropicProvider
-from exchange.tool import Tool
+from ai_exchange import Message, Text
+from ai_exchange.content import ToolResult, ToolUse
+from ai_exchange.providers.anthropic import AnthropicProvider
+from ai_exchange.tool import Tool
 
 
 def example_fn(param: str) -> None:

tests/providers/test_azure.py

@@ -2,8 +2,8 @@
 from unittest.mock import patch
 
 import pytest
-from exchange import Message, Text
-from exchange.providers.azure import AzureProvider
+from ai_exchange import Message, Text
+from ai_exchange.providers.azure import AzureProvider
 
 
 @pytest.fixture

tests/providers/test_bedrock.py

@@ -3,10 +3,10 @@
 from unittest.mock import patch
 
 import pytest
-from exchange.content import Text, ToolResult, ToolUse
-from exchange.message import Message
-from exchange.providers.bedrock import BedrockProvider
-from exchange.tool import Tool
+from ai_exchange.content import Text, ToolResult, ToolUse
+from ai_exchange.message import Message
+from ai_exchange.providers.bedrock import BedrockProvider
+from ai_exchange.tool import Tool
 
 logger = logging.getLogger(__name__)
 

tests/providers/test_databricks.py

@@ -2,8 +2,8 @@
 from unittest.mock import patch
 
 import pytest
-from exchange import Message, Text
-from exchange.providers.databricks import DatabricksProvider
+from ai_exchange import Message, Text
+from ai_exchange.providers.databricks import DatabricksProvider
 
 
 @pytest.fixture

tests/providers/test_openai.py

@@ -2,8 +2,8 @@
 from unittest.mock import patch
 
 import pytest
-from exchange import Message, Text
-from exchange.providers.openai import OpenAiProvider
+from ai_exchange import Message, Text
+from ai_exchange.providers.openai import OpenAiProvider
 
 
 @pytest.fixture

tests/providers/test_provider_utils.py

@@ -6,15 +6,15 @@
 import pytest
 from unittest.mock import patch
 
-from exchange.content import Text, ToolResult, ToolUse
-from exchange.message import Message
-from exchange.providers.utils import (
+from ai_exchange.content import Text, ToolResult, ToolUse
+from ai_exchange.message import Message
+from ai_exchange.providers.utils import (
     messages_to_openai_spec,
     openai_response_to_message,
     raise_for_status,
     tools_to_openai_spec,
 )
-from exchange.tool import Tool
+from ai_exchange.tool import Tool
 
 OPEN_AI_TOOL_USE_RESPONSE = response = {
     "choices": [

tests/test_exchange.py

@@ -1,11 +1,11 @@
 from typing import List
 
-from exchange.content import Text, ToolResult, ToolUse
-from exchange.exchange import Exchange
-from exchange.message import Message
-from exchange.moderators import PassiveModerator
-from exchange.providers import Provider, Usage
-from exchange.tool import Tool
+from ai_exchange.content import Text, ToolResult, ToolUse
+from ai_exchange.exchange import Exchange
+from ai_exchange.message import Message
+from ai_exchange.moderators import PassiveModerator
+from ai_exchange.providers import Provider, Usage
+from ai_exchange.tool import Tool
 
 
 def dummy_tool() -> str: