Revert "Updates dependencies and fixes errors and warnings" (#334)

Reverts #330

assistants/explorer-assistant/assistant/response/model.py

@@ -1,7 +1,7 @@
 from typing import Any, Protocol, Sequence
 
+from assistant_extensions.ai_clients.model import CompletionMessage
 from attr import dataclass
-from llm_client.model import CompletionMessage
 from semantic_workbench_api_model.workbench_model import (
     MessageType,
 )

assistants/explorer-assistant/assistant/response/response.py

@@ -11,9 +11,9 @@
 from typing import Any, Awaitable, Callable, Sequence
 
 import deepmerge
+from assistant_extensions.ai_clients.model import CompletionMessage
 from assistant_extensions.artifacts import ArtifactsExtension
 from assistant_extensions.attachments import AttachmentsExtension
-from llm_client.model import CompletionMessage
 from semantic_workbench_api_model.workbench_model import (
     ConversationMessage,
     ConversationParticipant,

assistants/explorer-assistant/assistant/response/response_anthropic.py

@@ -8,7 +8,7 @@
 from anthropic import NotGiven
 from anthropic.types import Message, MessageParam, TextBlock, ToolUseBlock
 from assistant_extensions.ai_clients.config import AnthropicClientConfigModel
-from llm_client.model import CompletionMessage
+from assistant_extensions.ai_clients.model import CompletionMessage
 from semantic_workbench_api_model.workbench_model import (
     MessageType,
 )

assistants/explorer-assistant/assistant/response/response_openai.py

@@ -6,8 +6,8 @@
 import deepmerge
 import openai_client
 from assistant_extensions.ai_clients.config import AzureOpenAIClientConfigModel, OpenAIClientConfigModel
+from assistant_extensions.ai_clients.model import CompletionMessage
 from assistant_extensions.artifacts import ArtifactsExtension
-from llm_client.model import CompletionMessage
 from openai.types.chat import (
     ChatCompletion,
     ChatCompletionDeveloperMessageParam,

assistants/prospector-assistant/assistant/chat.py

@@ -14,10 +14,10 @@
 
 import deepmerge
 import openai_client
+from assistant_extensions.ai_clients.model import CompletionMessageImageContent
 from assistant_extensions.attachments import AttachmentsExtension
 from content_safety.evaluators import CombinedContentSafetyEvaluator
 from openai.types.chat import ChatCompletionMessageParam
-from llm_client.model import CompletionMessageImageContent
 from pydantic import BaseModel, ConfigDict
 from semantic_workbench_api_model.workbench_model import (
     AssistantStateEvent,

examples/python/python-03-multimodel-chatbot/assistant/config.py

@@ -3,11 +3,11 @@
 from enum import StrEnum
 from typing import Annotated, Any, Literal
 
+import google.generativeai as genai
 import openai
 import openai_client
 from anthropic import AsyncAnthropic
 from content_safety.evaluators import CombinedContentSafetyEvaluatorConfig
-from google import genai
 from pydantic import BaseModel, ConfigDict, Field
 from semantic_workbench_assistant.config import ConfigSecretStr, UISchema
 
@@ -180,8 +180,9 @@ class GeminiServiceConfig(ServiceConfig):
         Field(title="Gemini Model", description="The Gemini model to use for generating responses."),
     ] = "gemini-1.5-pro"
 
-    def new_client(self, **kwargs) -> genai.Client:
-        return genai.Client(api_key=self.gemini_api_key)
+    def new_client(self, **kwargs) -> genai.GenerativeModel:
+        genai.configure(api_key=self.gemini_api_key)
+        return genai.GenerativeModel(self.gemini_model)
 
 
 # endregion

examples/python/python-03-multimodel-chatbot/assistant/model_adapters.py

@@ -6,7 +6,7 @@
 
 import anthropic
 import deepmerge
-from google.genai.types import Content
+import google.generativeai as genai
 from openai.types.chat import (
     ChatCompletion,
     ChatCompletionAssistantMessageParam,
@@ -16,7 +16,7 @@
     ChatCompletionUserMessageParam,
 )
 from openai.types.chat.chat_completion import Choice
-from pydantic import BaseModel, ConfigDict
+from pydantic import BaseModel
 
 from .config import (
     AnthropicServiceConfig,
@@ -42,7 +42,8 @@ class GenerateResponseResult(BaseModel):
 
 
 class ModelAdapter(BaseModel):
-    model_config = ConfigDict(arbitrary_types_allowed=True)
+    class Config:
+        arbitrary_types_allowed = True
 
     @abstractmethod
     def _format_messages(self, messages: List[Message]) -> Any:
@@ -229,7 +230,7 @@ async def generate_response(
 #
 
 
-GeminiFormattedMessages: TypeAlias = Iterable[Content]
+GeminiFormattedMessages: TypeAlias = Iterable[genai.types.StrictContentType]
 
 
 class GeminiAdapter(ModelAdapter):
@@ -264,12 +265,10 @@ async def generate_response(
         }
 
         try:
-            client = service_config.new_client()
-            chat = client.aio.chats.create(model=service_config.gemini_model, history=list(formatted_messages)[:-1])
-            latest_message = list(formatted_messages)[-1]
-            message = (latest_message.parts or [""])[0]
-            response = await chat.send_message(message)
-            deepmerge.always_merger.merge(metadata, {"debug": {"response": response.model_dump(mode="json")}})
+            model = service_config.new_client()
+            chat = model.start_chat(history=list(formatted_messages)[:-1])
+            response = await chat.send_message_async(list(formatted_messages)[-1])
+            deepmerge.always_merger.merge(metadata, {"debug": {"response": response.to_dict()}})
             return GenerateResponseResult(
                 response=response.text,
                 metadata=metadata,

examples/python/python-03-multimodel-chatbot/pyproject.toml

@@ -7,7 +7,7 @@ readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
     "anthropic>=0.34.2",
-    "google-genai>=1.2.0",
+    "google-generativeai>=0.8.1",
     "openai>=1.61.0",
     "tiktoken>=0.8.0",
     "semantic-workbench-assistant>=0.1.0",