change to ai_exchange

src/goose/build.py

@@ -1,8 +1,8 @@
 from itertools import chain
 
-from exchange import Exchange, Message
-from exchange.moderators import get_moderator
-from exchange.providers import get_provider
+from ai_exchange import Exchange, Message
+from ai_exchange.moderators import get_moderator
+from ai_exchange.providers import get_provider
 
 from goose.notifier import Notifier
 from goose.profile import Profile

src/goose/cli/main.py

@@ -23,7 +23,7 @@ def version() -> None:
     """Lists the version of goose and any plugins"""
     from importlib.metadata import entry_points, version
 
-    print(f"[green]Goose[/green]: [bold][cyan]{version('goose')}[/cyan][/bold]")
+    print(f"[green]Goose[/green]: [bold][cyan]{version('goose-ai')}[/cyan][/bold]")
     print("[green]Plugins[/green]:")
     filtered_groups = {}
     modules = set()

src/goose/cli/session.py

@@ -2,7 +2,7 @@
 from pathlib import Path
 from typing import Any, Dict, List, Optional
 
-from exchange import Message, ToolResult, ToolUse
+from ai_exchange import Message, ToolResult, ToolUse
 from prompt_toolkit.shortcuts import confirm
 from rich import print
 from rich.console import RenderableType

src/goose/toolkit/base.py

@@ -3,7 +3,7 @@
 from typing import Callable, Mapping, Optional, Tuple, TypeVar
 
 from attrs import define, field
-from exchange import Tool
+from ai_exchange import Tool
 
 from goose.notifier import Notifier
 

src/goose/toolkit/developer.py

@@ -2,7 +2,7 @@
 from subprocess import CompletedProcess, run
 from typing import List
 
-from exchange import Message
+from ai_exchange import Message
 from rich import box
 from rich.markdown import Markdown
 from rich.panel import Panel

src/goose/toolkit/github.py

@@ -1,4 +1,4 @@
-from exchange import Message
+from ai_exchange import Message
 
 from goose.toolkit.base import Toolkit
 

src/goose/toolkit/repo_context/repo_context.py

@@ -3,7 +3,7 @@
 from subprocess import CompletedProcess, run
 from typing import Dict, Tuple
 
-from exchange import Message
+from ai_exchange import Message
 
 from goose.notifier import Notifier
 from goose.toolkit import Toolkit

src/goose/toolkit/repo_context/utils.py

@@ -4,7 +4,7 @@
 from collections import deque
 from typing import Dict, List, Tuple
 
-from exchange import Exchange
+from ai_exchange import Exchange
 
 from goose.utils.ask import ask_an_ai
 

src/goose/toolkit/summarization/utils.py

@@ -4,8 +4,8 @@
 from pathlib import Path
 from typing import Dict, List, Optional, Tuple
 
-from exchange import Exchange
-from exchange.providers.utils import InitialMessageTooLargeError
+from ai_exchange import Exchange
+from ai_exchange.providers.utils import InitialMessageTooLargeError
 
 from goose.utils.ask import ask_an_ai
 from goose.utils.file_utils import create_file_list

src/goose/utils/ask.py

@@ -1,4 +1,4 @@
-from exchange import Exchange, Message
+from ai_exchange import Exchange, Message
 
 
 def ask_an_ai(input: str, exchange: Exchange, prompt: str = "", no_history: bool = True) -> Message:

src/goose/utils/session_file.py

@@ -2,7 +2,7 @@
 from pathlib import Path
 from typing import Dict, Iterator, List
 
-from exchange import Message
+from ai_exchange import Message
 
 from goose.cli.config import SESSION_FILE_SUFFIX
 

src/goose/view.py

@@ -1,5 +1,5 @@
 from attrs import define
-from exchange import Exchange
+from ai_exchange import Exchange
 
 
 @define

tests/cli/test_main.py

@@ -4,7 +4,7 @@
 
 import pytest
 from click.testing import CliRunner
-from exchange import Message
+from ai_exchange import Message
 from goose.cli.main import goose_cli
 
 

tests/cli/test_session.py

@@ -1,7 +1,7 @@
 from unittest.mock import MagicMock, patch
 
 import pytest
-from exchange import Message
+from ai_exchange import Message
 from goose.cli.prompt.goose_prompt_session import GoosePromptSession
 from goose.cli.prompt.user_input import PromptAction, UserInput
 from goose.cli.session import Session

tests/conftest.py

@@ -4,7 +4,7 @@
 from unittest.mock import Mock, patch
 
 import pytest
-from exchange import Exchange
+from ai_exchange import Exchange
 from goose.profile import Profile
 
 

tests/utils/test_ask.py

@@ -1,7 +1,7 @@
 from unittest.mock import MagicMock, patch
 
 import pytest
-from exchange import Exchange
+from ai_exchange import Exchange
 from goose.utils.ask import ask_an_ai, clear_exchange, replace_prompt
 
 

tests/utils/test_session_file.py

@@ -1,7 +1,7 @@
 from pathlib import Path
 
 import pytest
-from exchange import Message
+from ai_exchange import Message
 from goose.utils.session_file import list_sorted_session_files, read_from_file, session_file_exists, write_to_file