econagents 0.0.9__py3-none-any.whl → 0.0.10__py3-none-any.whl

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/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/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/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.dist-info → econagents-0.0.10.dist-info}/METADATA

@@ -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.dist-info → econagents-0.0.10.dist-info}/RECORD

@@ -1,4 +1,4 @@
-econagents/__init__.py,sha256=DBd48KXZ9WiiUNpLnp0NFTyFon9zb_RXNH1QQBmQyU4,1111
+econagents/__init__.py,sha256=FHKNa3fjxPTVwQGJMdJJ4OH3GaE7812_yGTIam5OVqs,1112
 econagents/_c_extension.pyi,sha256=evVvDNUCGqyMPrNViPF7QXfGUNNIMbUdY5HemRNQ1_o,113
 econagents/cli.py,sha256=Db2U9oSG9cyUv6UEM1wvWpDzo86MH8MpQmlEsk3vJU0,5394
 econagents/config_parser/__init__.py,sha256=jyVE581vqnBISyGidPh-MWX22brKB0wY-KM9rHPC4x4,408
@@ -17,13 +17,13 @@ econagents/core/state/fields.py,sha256=YxVOqdriaHRHoyeXsIB8ZDHygneMJD1OikOyeILK_
 econagents/core/state/game.py,sha256=6J3xV_SViShwBNgcWc_sK1dQPNO8Abb7qONFF0d4qVs,9729
 econagents/core/transport.py,sha256=dANw1BTwmUs99tb7PtCZSyJBQ23siiRZrBBmp2z5mbw,6427
 econagents/llm/__init__.py,sha256=J1PqpG3wL41oBAfzp5QaaP2ekwxCM_tiRq4H0OeSv4w,482
-econagents/llm/base.py,sha256=TFZhot8ffW-2JAOxDxeCG8V5Nn56ulSjCzJwSBopKg0,1726
+econagents/llm/base.py,sha256=lKjGf3bnzRq-hDSipASF7fX72jsEl1h1TV9AZp0eZiI,1703
 econagents/llm/observability.py,sha256=WSkJ7lZZl2FVpDWpFivudoeOENR_lyLZscUisveAi6k,9467
-econagents/llm/ollama.py,sha256=0ElnQhyCCAOu6u-TIC2JeNzh8OV6_VZjchFpi1F3HxU,2242
-econagents/llm/openai.py,sha256=x1L9GS2DXNvPTECXP6DVMwr9fuhcHMZh_2k8ds30i-Y,2395
+econagents/llm/ollama.py,sha256=RyneSeNKE1kC5mGE_i9rIUn8f4Kff0GHXEwOugCP6Jo,2408
+econagents/llm/openai.py,sha256=0mlg6HSuTfoM0nf3WTBuyEftqYQ3DRqfJvduVnkePW0,2561
 econagents/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-econagents-0.0.9.dist-info/LICENSE,sha256=Bd4MAEaMayyDO5BncOl3q0T2b6sWcjABigXLRFxgKIU,1082
-econagents-0.0.9.dist-info/METADATA,sha256=u3Db4fTyBEYZyv1SH2Dm0LJDoNZ0I6v6s37pjQEdvxE,4208
-econagents-0.0.9.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-econagents-0.0.9.dist-info/entry_points.txt,sha256=zm53n6FbaLAJuef_zRdddtTS-N0WTyWbxhcNUSdSZ0c,53
-econagents-0.0.9.dist-info/RECORD,,
+econagents-0.0.10.dist-info/LICENSE,sha256=Bd4MAEaMayyDO5BncOl3q0T2b6sWcjABigXLRFxgKIU,1082
+econagents-0.0.10.dist-info/METADATA,sha256=-MV1IafJATbKXJLIX69_Chj1pxVb0ZJb_Uhb-YmM7Ug,4209
+econagents-0.0.10.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+econagents-0.0.10.dist-info/entry_points.txt,sha256=zm53n6FbaLAJuef_zRdddtTS-N0WTyWbxhcNUSdSZ0c,53
+econagents-0.0.10.dist-info/RECORD,,