econagents 0.0.9__tar.gz → 0.0.10__tar.gz

{econagents-0.0.9 → econagents-0.0.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: econagents
-Version: 0.0.9
+Version: 0.0.10
 Summary: 
 License: MIT
 Author: Dylan Castillo

{econagents-0.0.9 → econagents-0.0.10}/econagents/__init__.py

@@ -12,7 +12,7 @@ from econagents.core.state.game import GameState, MetaInformation, PrivateInform
 from econagents.llm.openai import ChatOpenAI
 
 # Don't manually change, let poetry-dynamic-versioning handle it.
-__version__ = "0.0.9"
+__version__ = "0.0.10"
 
 __all__: list[str] = [
     "AgentRole",

{econagents-0.0.9 → econagents-0.0.10}/econagents/llm/base.py

@@ -47,7 +47,6 @@ class BaseLLM(ABC):
         self,
         messages: list[dict[str, Any]],
         tracing_extra: dict[str, Any],
-        **kwargs: Any,
     ) -> str:
         """Get a response from the LLM.
 

{econagents-0.0.9 → econagents-0.0.10}/econagents/llm/ollama.py

@@ -15,6 +15,7 @@ class ChatOllama(BaseLLM):
         self,
         model_name: str,
         host: Optional[str] = None,
+        response_kwargs: Optional[dict[str, Any]] = None,
     ) -> None:
         """Initialize the Ollama LLM interface.
 
@@ -25,17 +26,19 @@ class ChatOllama(BaseLLM):
         self._check_ollama_available()
         self.model_name = model_name
         self.host = host
+        self._response_kwargs = response_kwargs or {}
 
     def _check_ollama_available(self) -> None:
         """Check if Ollama is available."""
         if not importlib.util.find_spec("ollama"):
-            raise ImportError("Ollama is not installed. Install it with: pip install econagents[ollama]")
+            raise ImportError(
+                "Ollama is not installed. Install it with: pip install econagents[ollama]"
+            )
 
     async def get_response(
         self,
         messages: List[Dict[str, Any]],
         tracing_extra: Dict[str, Any],
-        **kwargs: Any,
     ) -> str:
         """Get a response from the LLM.
 
@@ -58,7 +61,7 @@ class ChatOllama(BaseLLM):
             response = await client.chat(
                 model=self.model_name,
                 messages=messages,
-                **kwargs,
+                **(self._response_kwargs),
             )
 
             # End the LLM run
@@ -74,4 +77,6 @@ class ChatOllama(BaseLLM):
 
         except ImportError as e:
             logger.error(f"Failed to import Ollama: {e}")
-            raise ImportError("Ollama is not installed. Install it with: pip install econagents[ollama]") from e
+            raise ImportError(
+                "Ollama is not installed. Install it with: pip install econagents[ollama]"
+            ) from e

{econagents-0.0.9 → econagents-0.0.10}/econagents/llm/openai.py

@@ -14,6 +14,7 @@ class ChatOpenAI(BaseLLM):
         self,
         model_name: str = "gpt-4o",
         api_key: Optional[str] = None,
+        response_kwargs: Optional[dict[str, Any]] = None,
     ) -> None:
         """Initialize the OpenAI LLM interface.
 
@@ -24,17 +25,19 @@ class ChatOpenAI(BaseLLM):
         self.model_name = model_name
         self.api_key = api_key
         self._check_openai_available()
+        self._response_kwargs = response_kwargs or {}
 
     def _check_openai_available(self) -> None:
         """Check if OpenAI is available."""
         if not importlib.util.find_spec("openai"):
-            raise ImportError("OpenAI is not installed. Install it with: pip install econagents[openai]")
+            raise ImportError(
+                "OpenAI is not installed. Install it with: pip install econagents[openai]"
+            )
 
     async def get_response(
         self,
         messages: list[dict[str, Any]],
         tracing_extra: dict[str, Any],
-        **kwargs: Any,
     ) -> str:
         """Get a response from the LLM.
 
@@ -59,7 +62,7 @@ class ChatOpenAI(BaseLLM):
                 model=self.model_name,
                 messages=messages,  # type: ignore
                 response_format={"type": "json_object"},
-                **kwargs,
+                **(self._response_kwargs),
             )
 
             # Track the LLM call using the observability provider
@@ -74,4 +77,6 @@ class ChatOpenAI(BaseLLM):
             return response.choices[0].message.content
         except ImportError as e:
             logger.error(f"Failed to import OpenAI: {e}")
-            raise ImportError("OpenAI is not installed. Install it with: pip install econagents[openai]") from e
+            raise ImportError(
+                "OpenAI is not installed. Install it with: pip install econagents[openai]"
+            ) from e

{econagents-0.0.9 → econagents-0.0.10}/pyproject.toml

@@ -27,7 +27,7 @@ dependencies = [
 
 [tool.poetry]
 name = "econagents"
-version = "0.0.9"  # Do not change, let poetry-dynamic-versioning handle it.
+version = "0.0.10"  # Do not change, let poetry-dynamic-versioning handle it.
 packages = [{include = "econagents"}]
 include = ["econagents/*.so", "econagents/*.pyd"]  # Compiled extensions
 license = "MIT"
@@ -65,10 +65,10 @@ pytest-mock = ">=3.7.0"
 python-dotenv = "^1.1.1"
 jupyter = "^1.1.1"
 nest-asyncio = "^1.6.0"
-ruff = "^0.12.7"
-types-requests = "^2.32.4.20250611"
+ruff = "^0.12.12"
+types-requests = "^2.32.4.20250809"
 pytest-asyncio = "^1.1.0"
-types-pyyaml = "^6.0.12.20250516"
+types-pyyaml = "^6.0.12.20250822"
 
 [tool.poetry.group.debug]
 optional = true