dm-aioaiagent 0.5.6__tar.gz → 0.6.0__tar.gz

{dm_aioaiagent-0.5.6/dm_aioaiagent.egg-info → dm_aioaiagent-0.6.0}/PKG-INFO

@@ -1,28 +1,46 @@
 Metadata-Version: 2.4
 Name: dm-aioaiagent
-Version: 0.5.6
+Version: 0.6.0
 Summary: This is my custom aioaiagent client
 Home-page: https://pypi.org/project/dm-aioaiagent
 Author: dimka4621
 Author-email: mismartconfig@gmail.com
 Project-URL: GitHub, https://github.com/MykhLibs/dm-aioaiagent
 Keywords: dm aioaiagent
-Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 Requires-Dist: dm-logger<0.7.0,>=0.6.6
-Requires-Dist: python-dotenv<1.1.0,>=1.0.0
+Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: pydantic<3.0.0,>=2.9.2
 Requires-Dist: langchain<0.4.0,>=0.3.0
 Requires-Dist: langchain-core<0.4.0,>=0.3.5
 Requires-Dist: langchain-community<0.4.0,>=0.3.0
 Requires-Dist: langchain-openai<0.4.0,>=0.3.0
-Requires-Dist: langchain-anthropic<0.4.0,>=0.3.0
 Requires-Dist: langgraph<0.4.0,>=0.3.23
 Requires-Dist: langsmith<0.4.0,>=0.3.45
 Requires-Dist: grandalf<0.9.0,>=0.8.0
+Provides-Extra: anthropic
+Requires-Dist: langchain-anthropic<0.4.0,>=0.3.0; extra == "anthropic"
+Provides-Extra: gemini
+Requires-Dist: langchain-google-genai<3.0.0,>=2.1.0; extra == "gemini"
+Provides-Extra: groq
+Requires-Dist: langchain-groq<0.4.0,>=0.3.0; extra == "groq"
+Provides-Extra: mistral
+Requires-Dist: langchain-mistralai<0.3.0,>=0.2.0; extra == "mistral"
+Provides-Extra: deepseek
+Requires-Dist: langchain-deepseek<0.2.0,>=0.1.0; extra == "deepseek"
+Provides-Extra: ollama
+Requires-Dist: langchain-ollama<0.4.0,>=0.3.0; extra == "ollama"
+Provides-Extra: all
+Requires-Dist: langchain-anthropic<0.4.0,>=0.3.0; extra == "all"
+Requires-Dist: langchain-google-genai<3.0.0,>=2.1.0; extra == "all"
+Requires-Dist: langchain-groq<0.4.0,>=0.3.0; extra == "all"
+Requires-Dist: langchain-mistralai<0.3.0,>=0.2.0; extra == "all"
+Requires-Dist: langchain-deepseek<0.2.0,>=0.1.0; extra == "all"
+Requires-Dist: langchain-ollama<0.4.0,>=0.3.0; extra == "all"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier
@@ -31,6 +49,7 @@ Dynamic: description-content-type
 Dynamic: home-page
 Dynamic: keywords
 Dynamic: project-url
+Dynamic: provides-extra
 Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary
@@ -44,6 +63,59 @@ Dynamic: summary
 
 ### * Package contains both `asynchronous` and `synchronous` clients
 
+## Installation
+
+By default, the package ships with **OpenAI** support. Other providers are optional extras:
+
+```bash
+pip install dm-aioaiagent                       # OpenAI only
+pip install dm-aioaiagent[anthropic]            # + Anthropic
+pip install dm-aioaiagent[anthropic,gemini]     # several at once
+pip install dm-aioaiagent[all]                  # every supported provider
+```
+
+Available extras: `anthropic`, `gemini`, `groq`, `mistral`, `deepseek`, `ollama`, `all`.
+
+If you call a model from a provider whose package is not installed, `init_chat_model` will raise an `ImportError` with the exact `pip install` command you need.
+
+## Providers
+
+Provider resolution is delegated to LangChain's [`init_chat_model`](https://python.langchain.com/api_reference/langchain/chat_models/langchain.chat_models.base.init_chat_model.html) — the agent picks the provider automatically by model name prefix when possible. For everything else, use the `"provider:model"` mask.
+
+```python
+# Auto-detected from model prefix (rules come from LangChain's init_chat_model)
+agent = DMAioAIAgent(model="gpt-4o-mini")              # → openai
+agent = DMAioAIAgent(model="claude-3-5-sonnet-latest") # → anthropic
+agent = DMAioAIAgent(model="gemini-2.0-flash")         # → google_vertexai (see note below)
+
+# Explicit provider via "provider:model" mask
+agent = DMAioAIAgent(model="google_genai:gemini-2.0-flash")
+agent = DMAioAIAgent(model="groq:llama-3.1-70b-versatile")
+agent = DMAioAIAgent(model="mistralai:mistral-large-latest")
+agent = DMAioAIAgent(model="deepseek:deepseek-chat")
+agent = DMAioAIAgent(model="ollama:llama3.1")
+
+# OpenAI-compatible gateway (OpenRouter, Together, vLLM, LiteLLM proxy, ...)
+# Works without installing any extra — just point to the OpenAI-compatible URL.
+agent = DMAioAIAgent(
+    model="meta-llama/llama-3.1-70b-instruct",
+    llm_provider_base_url="https://openrouter.ai/api/v1",
+    llm_provider_api_key="sk-or-...",
+)
+```
+
+> **Note about Gemini.** LangChain's auto-detect maps the `gemini*` prefix to **`google_vertexai`** (Google Cloud Vertex AI, requires a GCP service account). If you have a regular **Google AI Studio** API key (`GOOGLE_API_KEY`), use the `google_genai:` mask explicitly:
+>
+> ```python
+> agent = DMAioAIAgent(model="google_genai:gemini-2.0-flash")
+> ```
+
+Supported provider keys for the `"provider:model"` mask (list inherited from LangChain): `openai`, `anthropic`, `azure_openai`, `azure_ai`, `google_vertexai`, `google_genai`, `bedrock`, `bedrock_converse`, `cohere`, `fireworks`, `together`, `mistralai`, `huggingface`, `groq`, `ollama`, `google_anthropic_vertex`, `deepseek`, `ibm`, `nvidia`, `xai`, `perplexity`.
+
+### Note about parallel tool calls
+
+`parallel_tool_calls` is currently mapped only for **OpenAI** and **Anthropic** (their APIs use different formats). For other providers the parameter is silently ignored — extend per-provider mapping if you need it.
+
 ## Usage
 
 Analogue to `DMAioAIAgent` is the synchronous client `DMAIAgent`.
@@ -60,7 +132,7 @@ if sys.platform == "win32":
 
 ### Api Key Setup
 
-You can set your OpenAI API key in the environment variable `OPENAI_API_KEY` or pass it as an argument to the agent.
+Each provider reads its API key from a dedicated environment variable, e.g. `OPENAI_API_KEY`, `ANTHROPIC_API_KEY`, `GOOGLE_API_KEY`, `GROQ_API_KEY`, `MISTRAL_API_KEY`, etc. Alternatively, pass the key explicitly via the `llm_provider_api_key` argument — useful for multi-tenant setups, custom gateways, or runtime key rotation.
 
 **Use load_dotenv to load the `.env` file.**
 

dm_aioaiagent-0.5.6/PKG-INFO → dm_aioaiagent-0.6.0/README.md

@@ -1,40 +1,3 @@
-Metadata-Version: 2.4
-Name: dm-aioaiagent
-Version: 0.5.6
-Summary: This is my custom aioaiagent client
-Home-page: https://pypi.org/project/dm-aioaiagent
-Author: dimka4621
-Author-email: mismartconfig@gmail.com
-Project-URL: GitHub, https://github.com/MykhLibs/dm-aioaiagent
-Keywords: dm aioaiagent
-Classifier: Programming Language :: Python :: 3.8
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.9
-Description-Content-Type: text/markdown
-Requires-Dist: dm-logger<0.7.0,>=0.6.6
-Requires-Dist: python-dotenv<1.1.0,>=1.0.0
-Requires-Dist: pydantic<3.0.0,>=2.9.2
-Requires-Dist: langchain<0.4.0,>=0.3.0
-Requires-Dist: langchain-core<0.4.0,>=0.3.5
-Requires-Dist: langchain-community<0.4.0,>=0.3.0
-Requires-Dist: langchain-openai<0.4.0,>=0.3.0
-Requires-Dist: langchain-anthropic<0.4.0,>=0.3.0
-Requires-Dist: langgraph<0.4.0,>=0.3.23
-Requires-Dist: langsmith<0.4.0,>=0.3.45
-Requires-Dist: grandalf<0.9.0,>=0.8.0
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: keywords
-Dynamic: project-url
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
-
 # DM-aioaiagent
 
 ## Urls
@@ -44,6 +7,59 @@ Dynamic: summary
 
 ### * Package contains both `asynchronous` and `synchronous` clients
 
+## Installation
+
+By default, the package ships with **OpenAI** support. Other providers are optional extras:
+
+```bash
+pip install dm-aioaiagent                       # OpenAI only
+pip install dm-aioaiagent[anthropic]            # + Anthropic
+pip install dm-aioaiagent[anthropic,gemini]     # several at once
+pip install dm-aioaiagent[all]                  # every supported provider
+```
+
+Available extras: `anthropic`, `gemini`, `groq`, `mistral`, `deepseek`, `ollama`, `all`.
+
+If you call a model from a provider whose package is not installed, `init_chat_model` will raise an `ImportError` with the exact `pip install` command you need.
+
+## Providers
+
+Provider resolution is delegated to LangChain's [`init_chat_model`](https://python.langchain.com/api_reference/langchain/chat_models/langchain.chat_models.base.init_chat_model.html) — the agent picks the provider automatically by model name prefix when possible. For everything else, use the `"provider:model"` mask.
+
+```python
+# Auto-detected from model prefix (rules come from LangChain's init_chat_model)
+agent = DMAioAIAgent(model="gpt-4o-mini")              # → openai
+agent = DMAioAIAgent(model="claude-3-5-sonnet-latest") # → anthropic
+agent = DMAioAIAgent(model="gemini-2.0-flash")         # → google_vertexai (see note below)
+
+# Explicit provider via "provider:model" mask
+agent = DMAioAIAgent(model="google_genai:gemini-2.0-flash")
+agent = DMAioAIAgent(model="groq:llama-3.1-70b-versatile")
+agent = DMAioAIAgent(model="mistralai:mistral-large-latest")
+agent = DMAioAIAgent(model="deepseek:deepseek-chat")
+agent = DMAioAIAgent(model="ollama:llama3.1")
+
+# OpenAI-compatible gateway (OpenRouter, Together, vLLM, LiteLLM proxy, ...)
+# Works without installing any extra — just point to the OpenAI-compatible URL.
+agent = DMAioAIAgent(
+    model="meta-llama/llama-3.1-70b-instruct",
+    llm_provider_base_url="https://openrouter.ai/api/v1",
+    llm_provider_api_key="sk-or-...",
+)
+```
+
+> **Note about Gemini.** LangChain's auto-detect maps the `gemini*` prefix to **`google_vertexai`** (Google Cloud Vertex AI, requires a GCP service account). If you have a regular **Google AI Studio** API key (`GOOGLE_API_KEY`), use the `google_genai:` mask explicitly:
+>
+> ```python
+> agent = DMAioAIAgent(model="google_genai:gemini-2.0-flash")
+> ```
+
+Supported provider keys for the `"provider:model"` mask (list inherited from LangChain): `openai`, `anthropic`, `azure_openai`, `azure_ai`, `google_vertexai`, `google_genai`, `bedrock`, `bedrock_converse`, `cohere`, `fireworks`, `together`, `mistralai`, `huggingface`, `groq`, `ollama`, `google_anthropic_vertex`, `deepseek`, `ibm`, `nvidia`, `xai`, `perplexity`.
+
+### Note about parallel tool calls
+
+`parallel_tool_calls` is currently mapped only for **OpenAI** and **Anthropic** (their APIs use different formats). For other providers the parameter is silently ignored — extend per-provider mapping if you need it.
+
 ## Usage
 
 Analogue to `DMAioAIAgent` is the synchronous client `DMAIAgent`.
@@ -60,7 +76,7 @@ if sys.platform == "win32":
 
 ### Api Key Setup
 
-You can set your OpenAI API key in the environment variable `OPENAI_API_KEY` or pass it as an argument to the agent.
+Each provider reads its API key from a dedicated environment variable, e.g. `OPENAI_API_KEY`, `ANTHROPIC_API_KEY`, `GOOGLE_API_KEY`, `GROQ_API_KEY`, `MISTRAL_API_KEY`, etc. Alternatively, pass the key explicitly via the `llm_provider_api_key` argument — useful for multi-tenant setups, custom gateways, or runtime key rotation.
 
 **Use load_dotenv to load the `.env` file.**
 

{dm_aioaiagent-0.5.6 → dm_aioaiagent-0.6.0}/dm_aioaiagent/ai_agent.py

@@ -4,6 +4,7 @@ from typing import Any
 from pydantic import SecretStr
 from itertools import dropwhile
 from threading import Thread
+from langchain.chat_models import init_chat_model
 from langchain_core.tools import BaseTool
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_core.messages import SystemMessage, HumanMessage, AIMessage, ToolMessage
@@ -242,29 +243,28 @@ class DMAIAgent:
 
     def _init_agent(self) -> None:
         base_kwargs = {"model": self._model}
-        if isinstance(self._temperature, float):
-            base_kwargs["temperature"] = self._temperature
-        else:
-            ValueError("Temperature must be a float value.")
+        if self._temperature is not None:
+            if not isinstance(self._temperature, (int, float)):
+                raise ValueError("Temperature must be a float value.")
+            base_kwargs["temperature"] = float(self._temperature)
         if self._llm_provider_api_key:
             base_kwargs["api_key"] = SecretStr(self._llm_provider_api_key)
         if self._llm_provider_base_url:
             base_kwargs["base_url"] = self._llm_provider_base_url
 
-        if self._model.startswith("claude"):
-            from langchain_anthropic import ChatAnthropic
+        llm = init_chat_model(**base_kwargs)
 
-            llm = ChatAnthropic(**base_kwargs)
+        provider = self._detect_provider(self._model)
+        if provider == "anthropic":
             bind_tool_kwargs = {"tool_choice": {"type": "auto"}}
             if isinstance(self._parallel_tool_calls, bool):
                 bind_tool_kwargs["tool_choice"]["disable_parallel_tool_use"] = not self._parallel_tool_calls
-        else:
-            from langchain_openai import ChatOpenAI
-
-            llm = ChatOpenAI(**base_kwargs)
+        elif provider == "openai":
             bind_tool_kwargs = {}
             if isinstance(self._parallel_tool_calls, bool):
                 bind_tool_kwargs["parallel_tool_calls"] = self._parallel_tool_calls
+        else:
+            bind_tool_kwargs = {}
 
         if self._is_tools_exists:
             self._tool_map = {t.name: t for t in self._tools}
@@ -277,6 +277,16 @@ class DMAIAgent:
                                                    MessagesPlaceholder(variable_name="messages")])
         self._agent = prompt | llm
 
+    @staticmethod
+    def _detect_provider(model: str) -> str:
+        if ":" in model:
+            return model.split(":", 1)[0].replace("-", "_").lower()
+        if model.startswith(("gpt-", "o1", "o3")):
+            return "openai"
+        if model.startswith("claude"):
+            return "anthropic"
+        return ""
+
     def _init_graph(self) -> None:
         workflow = StateGraph(State)
         workflow.add_node("Prepare messages", self._prepare_messages_node)

dm_aioaiagent-0.6.0/dm_aioaiagent.egg-info/PKG-INFO

@@ -0,0 +1,264 @@
+Metadata-Version: 2.4
+Name: dm-aioaiagent
+Version: 0.6.0
+Summary: This is my custom aioaiagent client
+Home-page: https://pypi.org/project/dm-aioaiagent
+Author: dimka4621
+Author-email: mismartconfig@gmail.com
+Project-URL: GitHub, https://github.com/MykhLibs/dm-aioaiagent
+Keywords: dm aioaiagent
+Classifier: Programming Language :: Python :: 3.9
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: dm-logger<0.7.0,>=0.6.6
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: pydantic<3.0.0,>=2.9.2
+Requires-Dist: langchain<0.4.0,>=0.3.0
+Requires-Dist: langchain-core<0.4.0,>=0.3.5
+Requires-Dist: langchain-community<0.4.0,>=0.3.0
+Requires-Dist: langchain-openai<0.4.0,>=0.3.0
+Requires-Dist: langgraph<0.4.0,>=0.3.23
+Requires-Dist: langsmith<0.4.0,>=0.3.45
+Requires-Dist: grandalf<0.9.0,>=0.8.0
+Provides-Extra: anthropic
+Requires-Dist: langchain-anthropic<0.4.0,>=0.3.0; extra == "anthropic"
+Provides-Extra: gemini
+Requires-Dist: langchain-google-genai<3.0.0,>=2.1.0; extra == "gemini"
+Provides-Extra: groq
+Requires-Dist: langchain-groq<0.4.0,>=0.3.0; extra == "groq"
+Provides-Extra: mistral
+Requires-Dist: langchain-mistralai<0.3.0,>=0.2.0; extra == "mistral"
+Provides-Extra: deepseek
+Requires-Dist: langchain-deepseek<0.2.0,>=0.1.0; extra == "deepseek"
+Provides-Extra: ollama
+Requires-Dist: langchain-ollama<0.4.0,>=0.3.0; extra == "ollama"
+Provides-Extra: all
+Requires-Dist: langchain-anthropic<0.4.0,>=0.3.0; extra == "all"
+Requires-Dist: langchain-google-genai<3.0.0,>=2.1.0; extra == "all"
+Requires-Dist: langchain-groq<0.4.0,>=0.3.0; extra == "all"
+Requires-Dist: langchain-mistralai<0.3.0,>=0.2.0; extra == "all"
+Requires-Dist: langchain-deepseek<0.2.0,>=0.1.0; extra == "all"
+Requires-Dist: langchain-ollama<0.4.0,>=0.3.0; extra == "all"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: project-url
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# DM-aioaiagent
+
+## Urls
+
+* [PyPI](https://pypi.org/project/dm-aioaiagent)
+* [GitHub](https://github.com/MykhLibs/dm-aioaiagent)
+
+### * Package contains both `asynchronous` and `synchronous` clients
+
+## Installation
+
+By default, the package ships with **OpenAI** support. Other providers are optional extras:
+
+```bash
+pip install dm-aioaiagent                       # OpenAI only
+pip install dm-aioaiagent[anthropic]            # + Anthropic
+pip install dm-aioaiagent[anthropic,gemini]     # several at once
+pip install dm-aioaiagent[all]                  # every supported provider
+```
+
+Available extras: `anthropic`, `gemini`, `groq`, `mistral`, `deepseek`, `ollama`, `all`.
+
+If you call a model from a provider whose package is not installed, `init_chat_model` will raise an `ImportError` with the exact `pip install` command you need.
+
+## Providers
+
+Provider resolution is delegated to LangChain's [`init_chat_model`](https://python.langchain.com/api_reference/langchain/chat_models/langchain.chat_models.base.init_chat_model.html) — the agent picks the provider automatically by model name prefix when possible. For everything else, use the `"provider:model"` mask.
+
+```python
+# Auto-detected from model prefix (rules come from LangChain's init_chat_model)
+agent = DMAioAIAgent(model="gpt-4o-mini")              # → openai
+agent = DMAioAIAgent(model="claude-3-5-sonnet-latest") # → anthropic
+agent = DMAioAIAgent(model="gemini-2.0-flash")         # → google_vertexai (see note below)
+
+# Explicit provider via "provider:model" mask
+agent = DMAioAIAgent(model="google_genai:gemini-2.0-flash")
+agent = DMAioAIAgent(model="groq:llama-3.1-70b-versatile")
+agent = DMAioAIAgent(model="mistralai:mistral-large-latest")
+agent = DMAioAIAgent(model="deepseek:deepseek-chat")
+agent = DMAioAIAgent(model="ollama:llama3.1")
+
+# OpenAI-compatible gateway (OpenRouter, Together, vLLM, LiteLLM proxy, ...)
+# Works without installing any extra — just point to the OpenAI-compatible URL.
+agent = DMAioAIAgent(
+    model="meta-llama/llama-3.1-70b-instruct",
+    llm_provider_base_url="https://openrouter.ai/api/v1",
+    llm_provider_api_key="sk-or-...",
+)
+```
+
+> **Note about Gemini.** LangChain's auto-detect maps the `gemini*` prefix to **`google_vertexai`** (Google Cloud Vertex AI, requires a GCP service account). If you have a regular **Google AI Studio** API key (`GOOGLE_API_KEY`), use the `google_genai:` mask explicitly:
+>
+> ```python
+> agent = DMAioAIAgent(model="google_genai:gemini-2.0-flash")
+> ```
+
+Supported provider keys for the `"provider:model"` mask (list inherited from LangChain): `openai`, `anthropic`, `azure_openai`, `azure_ai`, `google_vertexai`, `google_genai`, `bedrock`, `bedrock_converse`, `cohere`, `fireworks`, `together`, `mistralai`, `huggingface`, `groq`, `ollama`, `google_anthropic_vertex`, `deepseek`, `ibm`, `nvidia`, `xai`, `perplexity`.
+
+### Note about parallel tool calls
+
+`parallel_tool_calls` is currently mapped only for **OpenAI** and **Anthropic** (their APIs use different formats). For other providers the parameter is silently ignored — extend per-provider mapping if you need it.
+
+## Usage
+
+Analogue to `DMAioAIAgent` is the synchronous client `DMAIAgent`.
+
+### Windows Setup
+
+```python
+import asyncio
+import sys
+
+if sys.platform == "win32":
+    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
+```
+
+### Api Key Setup
+
+Each provider reads its API key from a dedicated environment variable, e.g. `OPENAI_API_KEY`, `ANTHROPIC_API_KEY`, `GOOGLE_API_KEY`, `GROQ_API_KEY`, `MISTRAL_API_KEY`, etc. Alternatively, pass the key explicitly via the `llm_provider_api_key` argument — useful for multi-tenant setups, custom gateways, or runtime key rotation.
+
+**Use load_dotenv to load the `.env` file.**
+
+```python
+from dotenv import load_dotenv
+load_dotenv()
+```
+
+### Use agent *with* inner memory and run *single* message
+
+By default, agent use inner memory to store the conversation history.
+
+(You can set *max count messages in memory* by `max_memory_messages` init argument)
+
+```python
+import asyncio
+from dm_aioaiagent import DMAioAIAgent
+
+
+async def main():
+    # define a system message
+    system_message = "Your custom system message with role, backstory and goal"
+
+    # (optional) define a list of tools, if you want to use them
+    tools = [...]
+
+    # define a openai model, default is "gpt-4o-mini"
+    model_name = "gpt-4o"
+
+    # create an agent
+    ai_agent = DMAioAIAgent(system_message, tools, model=model_name)
+    # if you don't want to see the input and output messages from agent
+    # you can set `input_output_logging=False` init argument
+
+    # call an agent
+    answer = await ai_agent.run("Hello!")
+
+    # call an agent
+    answer = await ai_agent.run("I want to know the weather in Kyiv")
+
+    # get full conversation history
+    conversation_history = ai_agent.memory_messages
+
+    # clear conversation history
+    ai_agent.clear_memory_messages()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+### Use agent *without* inner memory and run *multiple* messages
+
+If you want to control the memory of the agent, you can disable it by setting `is_memory_enabled=False`
+
+```python
+import asyncio
+from dm_aioaiagent import DMAioAIAgent
+
+
+async def main():
+    # define a system message
+    system_message = "Your custom system message with role, backstory and goal"
+
+    # (optional) define a list of tools, if you want to use them
+    tools = [...]
+
+    # define a openai model, default is "gpt-4o-mini"
+    model_name = "gpt-4o"
+
+    # create an agent
+    ai_agent = DMAioAIAgent(system_message, tools, model=model_name,
+                            is_memory_enabled=False)
+    # if you don't want to see the input and output messages from agent
+    # you can set input_output_logging=False
+
+    # define the conversation message(s)
+    messages = [
+        {"role": "user", "content": "Hello!"}
+    ]
+
+    # call an agent
+    new_messages = await ai_agent.run_messages(messages)
+
+    # add new_messages to messages
+    messages.extend(new_messages)
+
+    # define the next conversation message
+    messages.append(
+        {"role": "user", "content": "I want to know the weather in Kyiv"}
+    )
+
+    # call an agent
+    new_messages = await ai_agent.run_messages(messages)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+### Image vision
+
+```python
+from dm_aioaiagent import DMAIAgent, OpenAIImageMessageContent
+
+
+def main():
+    # create an agent
+    ai_agent = DMAIAgent(agent_name="image_vision", model="gpt-4o")
+
+    # create an image message content
+    # NOTE: text argument is optional
+    img_content = OpenAIImageMessageContent(image_url="https://your.domain/image",
+                                            text="Hello, what is shown in the photo?")
+
+    # define the conversation messages
+    messages = [
+        {"role": "user", "content": "Hello!"},
+        {"role": "user", "content": img_content},
+    ]
+
+    # call an agent
+    new_messages = ai_agent.run_messages(messages)
+    answer = new_messages[-1].content
+
+
+if __name__ == "__main__":
+    main()
+```

dm_aioaiagent-0.6.0/dm_aioaiagent.egg-info/requires.txt

@@ -0,0 +1,36 @@
+dm-logger<0.7.0,>=0.6.6
+python-dotenv>=1.0.0
+pydantic<3.0.0,>=2.9.2
+langchain<0.4.0,>=0.3.0
+langchain-core<0.4.0,>=0.3.5
+langchain-community<0.4.0,>=0.3.0
+langchain-openai<0.4.0,>=0.3.0
+langgraph<0.4.0,>=0.3.23
+langsmith<0.4.0,>=0.3.45
+grandalf<0.9.0,>=0.8.0
+
+[all]
+langchain-anthropic<0.4.0,>=0.3.0
+langchain-google-genai<3.0.0,>=2.1.0
+langchain-groq<0.4.0,>=0.3.0
+langchain-mistralai<0.3.0,>=0.2.0
+langchain-deepseek<0.2.0,>=0.1.0
+langchain-ollama<0.4.0,>=0.3.0
+
+[anthropic]
+langchain-anthropic<0.4.0,>=0.3.0
+
+[deepseek]
+langchain-deepseek<0.2.0,>=0.1.0
+
+[gemini]
+langchain-google-genai<3.0.0,>=2.1.0
+
+[groq]
+langchain-groq<0.4.0,>=0.3.0
+
+[mistral]
+langchain-mistralai<0.3.0,>=0.2.0
+
+[ollama]
+langchain-ollama<0.4.0,>=0.3.0

{dm_aioaiagent-0.5.6 → dm_aioaiagent-0.6.0}/setup.py

@@ -8,7 +8,7 @@ def readme():
 
 setup(
     name='dm-aioaiagent',
-    version='v0.5.6',
+    version='v0.6.0',
     author='dimka4621',
     author_email='mismartconfig@gmail.com',
     description='This is my custom aioaiagent client',
@@ -18,19 +18,34 @@ setup(
     packages=find_packages(),
     install_requires=[
         'dm-logger>=0.6.6, <0.7.0',
-        'python-dotenv>=1.0.0, <1.1.0',
+        'python-dotenv>=1.0.0',
         'pydantic>=2.9.2, <3.0.0',
         'langchain>=0.3.0, <0.4.0',
         'langchain-core>=0.3.5, <0.4.0',
         'langchain-community>=0.3.0, <0.4.0',
         'langchain-openai>=0.3.0, <0.4.0',
-        'langchain-anthropic>=0.3.0, <0.4.0',
         'langgraph>=0.3.23, <0.4.0',
         'langsmith>=0.3.45, <0.4.0',
         'grandalf>=0.8.0, <0.9.0',
     ],
+    extras_require={
+        'anthropic': ['langchain-anthropic>=0.3.0, <0.4.0'],
+        'gemini':    ['langchain-google-genai>=2.1.0, <3.0.0'],
+        'groq':      ['langchain-groq>=0.3.0, <0.4.0'],
+        'mistral':   ['langchain-mistralai>=0.2.0, <0.3.0'],
+        'deepseek':  ['langchain-deepseek>=0.1.0, <0.2.0'],
+        'ollama':    ['langchain-ollama>=0.3.0, <0.4.0'],
+        'all': [
+            'langchain-anthropic>=0.3.0, <0.4.0',
+            'langchain-google-genai>=2.1.0, <3.0.0',
+            'langchain-groq>=0.3.0, <0.4.0',
+            'langchain-mistralai>=0.2.0, <0.3.0',
+            'langchain-deepseek>=0.1.0, <0.2.0',
+            'langchain-ollama>=0.3.0, <0.4.0',
+        ],
+    },
     classifiers=[
-        'Programming Language :: Python :: 3.8',
+        'Programming Language :: Python :: 3.9',
         'License :: OSI Approved :: MIT License',
         'Operating System :: OS Independent'
     ],

dm_aioaiagent-0.5.6/README.md

@@ -1,155 +0,0 @@
-# DM-aioaiagent
-
-## Urls
-
-* [PyPI](https://pypi.org/project/dm-aioaiagent)
-* [GitHub](https://github.com/MykhLibs/dm-aioaiagent)
-
-### * Package contains both `asynchronous` and `synchronous` clients
-
-## Usage
-
-Analogue to `DMAioAIAgent` is the synchronous client `DMAIAgent`.
-
-### Windows Setup
-
-```python
-import asyncio
-import sys
-
-if sys.platform == "win32":
-    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
-```
-
-### Api Key Setup
-
-You can set your OpenAI API key in the environment variable `OPENAI_API_KEY` or pass it as an argument to the agent.
-
-**Use load_dotenv to load the `.env` file.**
-
-```python
-from dotenv import load_dotenv
-load_dotenv()
-```
-
-### Use agent *with* inner memory and run *single* message
-
-By default, agent use inner memory to store the conversation history.
-
-(You can set *max count messages in memory* by `max_memory_messages` init argument)
-
-```python
-import asyncio
-from dm_aioaiagent import DMAioAIAgent
-
-
-async def main():
-    # define a system message
-    system_message = "Your custom system message with role, backstory and goal"
-
-    # (optional) define a list of tools, if you want to use them
-    tools = [...]
-
-    # define a openai model, default is "gpt-4o-mini"
-    model_name = "gpt-4o"
-
-    # create an agent
-    ai_agent = DMAioAIAgent(system_message, tools, model=model_name)
-    # if you don't want to see the input and output messages from agent
-    # you can set `input_output_logging=False` init argument
-
-    # call an agent
-    answer = await ai_agent.run("Hello!")
-
-    # call an agent
-    answer = await ai_agent.run("I want to know the weather in Kyiv")
-
-    # get full conversation history
-    conversation_history = ai_agent.memory_messages
-
-    # clear conversation history
-    ai_agent.clear_memory_messages()
-
-
-if __name__ == "__main__":
-    asyncio.run(main())
-```
-
-### Use agent *without* inner memory and run *multiple* messages
-
-If you want to control the memory of the agent, you can disable it by setting `is_memory_enabled=False`
-
-```python
-import asyncio
-from dm_aioaiagent import DMAioAIAgent
-
-
-async def main():
-    # define a system message
-    system_message = "Your custom system message with role, backstory and goal"
-
-    # (optional) define a list of tools, if you want to use them
-    tools = [...]
-
-    # define a openai model, default is "gpt-4o-mini"
-    model_name = "gpt-4o"
-
-    # create an agent
-    ai_agent = DMAioAIAgent(system_message, tools, model=model_name,
-                            is_memory_enabled=False)
-    # if you don't want to see the input and output messages from agent
-    # you can set input_output_logging=False
-
-    # define the conversation message(s)
-    messages = [
-        {"role": "user", "content": "Hello!"}
-    ]
-
-    # call an agent
-    new_messages = await ai_agent.run_messages(messages)
-
-    # add new_messages to messages
-    messages.extend(new_messages)
-
-    # define the next conversation message
-    messages.append(
-        {"role": "user", "content": "I want to know the weather in Kyiv"}
-    )
-
-    # call an agent
-    new_messages = await ai_agent.run_messages(messages)
-
-
-if __name__ == "__main__":
-    asyncio.run(main())
-```
-
-### Image vision
-
-```python
-from dm_aioaiagent import DMAIAgent, OpenAIImageMessageContent
-
-
-def main():
-    # create an agent
-    ai_agent = DMAIAgent(agent_name="image_vision", model="gpt-4o")
-
-    # create an image message content
-    # NOTE: text argument is optional
-    img_content = OpenAIImageMessageContent(image_url="https://your.domain/image",
-                                            text="Hello, what is shown in the photo?")
-
-    # define the conversation messages
-    messages = [
-        {"role": "user", "content": "Hello!"},
-        {"role": "user", "content": img_content},
-    ]
-
-    # call an agent
-    new_messages = ai_agent.run_messages(messages)
-    answer = new_messages[-1].content
-
-
-if __name__ == "__main__":
-    main()
-```

dm_aioaiagent-0.5.6/dm_aioaiagent.egg-info/requires.txt

@@ -1,11 +0,0 @@
-dm-logger<0.7.0,>=0.6.6
-python-dotenv<1.1.0,>=1.0.0
-pydantic<3.0.0,>=2.9.2
-langchain<0.4.0,>=0.3.0
-langchain-core<0.4.0,>=0.3.5
-langchain-community<0.4.0,>=0.3.0
-langchain-openai<0.4.0,>=0.3.0
-langchain-anthropic<0.4.0,>=0.3.0
-langgraph<0.4.0,>=0.3.23
-langsmith<0.4.0,>=0.3.45
-grandalf<0.9.0,>=0.8.0