openrouter-provider 1.0.5__py3-none-any.whl → 1.0.6__py3-none-any.whl

openrouter/openrouter.py

@@ -82,45 +82,54 @@ class OpenRouterClient:
         model: LLMModel,
         query: Message,
         tools: list[tool_model] = None,
+        max_tools_loop: int = 3,
         provider: ProviderConfig = None,
         temperature: float = 0.3
     ) -> Message:
         tools = tools or []
         self._memory.append(query)
         client = OpenRouterProvider()
-        reply = client.invoke(
-            model=model,
-            temperature=temperature,
-            system_prompt=self._system_prompt,
-            querys=self._memory,
-            tools=self.tools + tools,
-            provider=provider,
-        )
-        reply.answered_by = model
-        self._memory.append(reply)
-
-        if reply.tool_calls:
-            for requested_tool in reply.tool_calls:
-                args = requested_tool.arguments
-                if isinstance(args, str):
-                    args = json.loads(args)
-
-                for tool in (self.tools + tools):
-                    if tool.name == requested_tool.name:
-                        result = tool(**args)
-                        requested_tool.result = result
-                        break
-                else:
-                    return reply
-                    
+        
+        for i in range(max_tools_loop):
             reply = client.invoke(
                 model=model,
+                temperature=temperature,
                 system_prompt=self._system_prompt,
                 querys=self._memory,
                 tools=self.tools + tools,
-                provider=provider
+                provider=provider,
             )
+            reply.answered_by = model
+            self._memory.append(reply)
+
+            # Execute tools
+            if reply.tool_calls:
+                for requested_tool in reply.tool_calls:
+                    args = requested_tool.arguments
+                    if isinstance(args, str):
+                        args = json.loads(args)
+
+                    for tool in (self.tools + tools):
+                        if tool.name == requested_tool.name:
+                            result = tool(**args)
+                            requested_tool.result = result
+                            break
+                    else:
+                        return reply
             
+            # If normal reply, do nothing
+            else:           
+                break
+        
+        # If over max execution, send message without tools
+        else:
+            reply = client.invoke(
+                model=model,
+                temperature=temperature,
+                system_prompt=self._system_prompt,
+                querys=self._memory,
+                provider=provider,
+            )
             reply.answered_by = model
             self._memory.append(reply)
 

{openrouter_provider-1.0.5.dist-info → openrouter_provider-1.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrouter-provider
-Version: 1.0.5
+Version: 1.0.6
 Summary: This is an unofficial wrapper of OpenRouter.
 Author-email: Keisuke Miyamto <aichiboyhighschool@gmail.com>
 Requires-Python: >=3.7

{openrouter_provider-1.0.5.dist-info → openrouter_provider-1.0.6.dist-info}/RECORD

@@ -1,10 +1,10 @@
 openrouter/__init__.py,sha256=xuIzdm8-l3Tmc-zrNIXTicv05c9HCMxTS9xynKpWK-Q,123
 openrouter/llms.py,sha256=zmujFW5BmQA0Fm6z8skuO3ipLpaUMy1VBZxYkPE9OcM,3391
 openrouter/message.py,sha256=0q0OYdTl3f1TOx79TJRwPYyFuTxoYI9tSCpQN-N8G-A,2870
-openrouter/openrouter.py,sha256=wqDTos1_o2db-Fs4ACbwwx5CuOLl3JvJpGHpW5HIjJ0,8015
+openrouter/openrouter.py,sha256=J74xqt3ycpD_RUhScEzX3vKc4zfwaJE6Wq3SLGDO6Kc,8368
 openrouter/openrouter_provider.py,sha256=w6isS1VX7iMf71zwQ5tMv7v5X0yeMbLF6vqimOrB724,8567
 openrouter/tool.py,sha256=tUUNLosz1XhzPIwY1zHXWNM3ePs7hcVD1a_W5hWTCWk,1975
-openrouter_provider-1.0.5.dist-info/METADATA,sha256=Vk8scE87sGlt_hAMlfRLQRpsluzvYqvgOaR2_xjWQZI,11503
-openrouter_provider-1.0.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-openrouter_provider-1.0.5.dist-info/top_level.txt,sha256=0jnlCcRirGeYZLm5ZbWQRUonIp4tTPl_9mq-ds_1SEo,11
-openrouter_provider-1.0.5.dist-info/RECORD,,
+openrouter_provider-1.0.6.dist-info/METADATA,sha256=92TvxKqv7j2j3s0kahWhQKe3pc6VUv5t6-h7gH4onrs,11503
+openrouter_provider-1.0.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+openrouter_provider-1.0.6.dist-info/top_level.txt,sha256=0jnlCcRirGeYZLm5ZbWQRUonIp4tTPl_9mq-ds_1SEo,11
+openrouter_provider-1.0.6.dist-info/RECORD,,