dm-aioaiagent 0.4.2__py3-none-any.whl → 0.4.5__py3-none-any.whl

dm_aioaiagent/ai_agent.py

@@ -33,6 +33,7 @@ class DMAIAgent:
         max_memory_messages: int = MAX_MEMORY_MESSAGES,
         save_tools_responses_in_memory: bool = True,
         llm_provider_api_key: str = "",
+        llm_provider_base_url: str = "",
         response_if_request_fail: str = "I can't provide a response right now. Please try again later.",
         response_if_invalid_image: str = "The image is unavailable or the link is incorrect."
     ):
@@ -46,6 +47,7 @@ class DMAIAgent:
         self._temperature = int(temperature)
         self._parallel_tool_calls = bool(parallel_tool_calls)
         self._llm_provider_api_key = str(llm_provider_api_key)
+        self._llm_provider_base_url = str(llm_provider_base_url)
 
         self._memory_messages = []
         self._is_memory_enabled = bool(is_memory_enabled)
@@ -125,7 +127,9 @@ class DMAIAgent:
             self._logger.error(e)
             if second_attempt:
                 response = self._response_if_invalid_image if "invalid_image_url" in str(e) else self._response_if_request_fail
-                state["messages"].append(AIMessage(content=response))
+                ai_response = AIMessage(content=response)
+                state["messages"].append(ai_response)
+                state["new_messages"].append(ai_response)
                 return state
             return self._invoke_llm_node(state, second_attempt=True)
         state["messages"].append(ai_response)
@@ -189,21 +193,22 @@ class DMAIAgent:
         return route
 
     def _init_agent(self) -> None:
+        base_kwargs = {
+            "model": self._model,
+            "temperature": self._temperature,
+            "base_url": self._llm_provider_base_url if self._llm_provider_base_url else None
+        }
         if self._llm_provider_api_key:
-            self._llm_provider_api_key = SecretStr(self._llm_provider_api_key)
-
-        if self._model.startswith("gpt"):
-            from langchain_openai import ChatOpenAI
+            base_kwargs["api_key"] = SecretStr(self._llm_provider_api_key)
 
-            api_key = SecretStr(self._llm_provider_api_key or os.getenv("OPENAI_API_KEY"))
-            llm = ChatOpenAI(model_name=self._model, temperature=self._temperature, openai_api_key=api_key)
-        elif self._model.startswith("claude"):
+        if self._model.startswith("claude"):
             from langchain_anthropic import ChatAnthropic
 
-            api_key = SecretStr(self._llm_provider_api_key or os.getenv("ANTHROPIC_API_KEY"))
-            llm = ChatAnthropic(model=self._model, temperature=self._temperature, anthropic_api_key=api_key)
+            llm = ChatAnthropic(**base_kwargs)
         else:
-            raise ValueError(f"{self.__class__.__name__} not support this model: '{self._model}'")
+            from langchain_openai import ChatOpenAI
+
+            llm = ChatOpenAI(**base_kwargs)
 
         if self._is_tools_exists:
             self._tool_map = {t.name: t for t in self._tools}

dm_aioaiagent/async_ai_agent.py

@@ -51,7 +51,9 @@ class DMAioAIAgent(DMAIAgent):
             self._logger.error(e)
             if second_attempt:
                 response = self._response_if_invalid_image if "invalid_image_url" in str(e) else self._response_if_request_fail
-                state["messages"].append(AIMessage(content=response))
+                ai_response = AIMessage(content=response)
+                state["messages"].append(ai_response)
+                state["new_messages"].append(ai_response)
                 return state
             return await self._invoke_llm_node(state, second_attempt=True)
         state["messages"].append(ai_response)

{dm_aioaiagent-0.4.2.dist-info → dm_aioaiagent-0.4.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: dm-aioaiagent
-Version: 0.4.2
+Version: 0.4.5
 Summary: This is my custom aioaiagent client
 Home-page: https://pypi.org/project/dm-aioaiagent
 Author: dimka4621
@@ -17,12 +17,12 @@ Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: pydantic<3.0.0,>=2.9.2
 Requires-Dist: langchain~=0.3.0
 Requires-Dist: langchain-core~=0.3.5
+Requires-Dist: langchain-community~=0.3.0
+Requires-Dist: langchain-openai~=0.3.0
+Requires-Dist: langchain-anthropic~=0.3.0
 Requires-Dist: langgraph~=0.2.23
 Requires-Dist: langsmith~=0.1.144
 Requires-Dist: grandalf>=0.8
-Requires-Dist: langchain-community~=0.3.0
-Requires-Dist: langchain-openai~=0.2.0
-Requires-Dist: langchain-anthropic~=0.3.0
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

dm_aioaiagent-0.4.5.dist-info/RECORD

@@ -0,0 +1,9 @@
+dm_aioaiagent/__init__.py,sha256=4Lkfntw5JnPyWvCppc-NbRznB2LRA2J-afJK8T4mGQs,186
+dm_aioaiagent/ai_agent.py,sha256=o8GRFwYYEFJYBAyWAQkSZQT0Ou73TcrL-AKZlAQmuZE,11112
+dm_aioaiagent/async_ai_agent.py,sha256=DArXNs2FEMHgZA7sURpnuPmujbsoDA1zSpPkKT0HQW4,3674
+dm_aioaiagent/openai_image_message_content.py,sha256=EP_i0ERCz7c4KOM8UXp2-AI91ntGC7PGQBc5MMNspcs,434
+dm_aioaiagent/types.py,sha256=ZUfC7eEh7iHSceFy1EscM7D4OohZ-QkdfOAgKP4EaJk,663
+dm_aioaiagent-0.4.5.dist-info/METADATA,sha256=1ceoqrZwa5BdjLD2QMNNkc2akhjpFvt_hDXFhRcNymI,5141
+dm_aioaiagent-0.4.5.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+dm_aioaiagent-0.4.5.dist-info/top_level.txt,sha256=CbasLH0KI7zA77XwT6JDCnmRascxKNGvUVV9MgYjHAU,14
+dm_aioaiagent-0.4.5.dist-info/RECORD,,

dm_aioaiagent-0.4.2.dist-info/RECORD

@@ -1,9 +0,0 @@
-dm_aioaiagent/__init__.py,sha256=4Lkfntw5JnPyWvCppc-NbRznB2LRA2J-afJK8T4mGQs,186
-dm_aioaiagent/ai_agent.py,sha256=AbUx8VdznO5irFT3t0Fk1PenPqVIiJgKfZReTx9IS4Y,11157
-dm_aioaiagent/async_ai_agent.py,sha256=egaK2CtHE66bxbPzounseDZgJYsfTg3DAPi1zcmc6RQ,3574
-dm_aioaiagent/openai_image_message_content.py,sha256=EP_i0ERCz7c4KOM8UXp2-AI91ntGC7PGQBc5MMNspcs,434
-dm_aioaiagent/types.py,sha256=ZUfC7eEh7iHSceFy1EscM7D4OohZ-QkdfOAgKP4EaJk,663
-dm_aioaiagent-0.4.2.dist-info/METADATA,sha256=sExdT6znaLzX6AvSfNsc6So-Zn-XAOJJ9BdnH5ScahU,5141
-dm_aioaiagent-0.4.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-dm_aioaiagent-0.4.2.dist-info/top_level.txt,sha256=CbasLH0KI7zA77XwT6JDCnmRascxKNGvUVV9MgYjHAU,14
-dm_aioaiagent-0.4.2.dist-info/RECORD,,