langroid 0.1.240__tar.gz → 0.1.241__tar.gz

{langroid-0.1.240 → langroid-0.1.241}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langroid
-Version: 0.1.240
+Version: 0.1.241
 Summary: Harness LLMs with Multi-Agent Programming
 License: MIT
 Author: Prasad Chalasani
@@ -36,6 +36,7 @@ Requires-Dist: fakeredis (>=2.12.1,<3.0.0)
 Requires-Dist: fire (>=0.5.0,<0.6.0)
 Requires-Dist: flake8 (>=6.0.0,<7.0.0)
 Requires-Dist: google-api-python-client (>=2.95.0,<3.0.0)
+Requires-Dist: google-generativeai (>=0.5.2,<0.6.0)
 Requires-Dist: groq (>=0.5.0,<0.6.0)
 Requires-Dist: grpcio (>=1.62.1,<2.0.0)
 Requires-Dist: halo (>=0.0.31,<0.0.32)

{langroid-0.1.240 → langroid-0.1.241}/langroid/language_models/openai_gpt.py

@@ -670,23 +670,26 @@ class OpenAIGPT(LanguageModel):
         sys.stdout.write(Colors().GREEN)
         sys.stdout.flush()
         has_function = False
-        for event in response:
-            (
-                is_break,
-                has_function,
-                function_name,
-                function_args,
-                completion,
-            ) = self._process_stream_event(
-                event,
-                chat=chat,
-                has_function=has_function,
-                completion=completion,
-                function_args=function_args,
-                function_name=function_name,
-            )
-            if is_break:
-                break
+        try:
+            for event in response:
+                (
+                    is_break,
+                    has_function,
+                    function_name,
+                    function_args,
+                    completion,
+                ) = self._process_stream_event(
+                    event,
+                    chat=chat,
+                    has_function=has_function,
+                    completion=completion,
+                    function_args=function_args,
+                    function_name=function_name,
+                )
+                if is_break:
+                    break
+        except Exception:
+            pass
 
         print("")
         # TODO- get usage info in stream mode (?)
@@ -722,23 +725,26 @@ class OpenAIGPT(LanguageModel):
         sys.stdout.write(Colors().GREEN)
         sys.stdout.flush()
         has_function = False
-        async for event in response:
-            (
-                is_break,
-                has_function,
-                function_name,
-                function_args,
-                completion,
-            ) = self._process_stream_event(
-                event,
-                chat=chat,
-                has_function=has_function,
-                completion=completion,
-                function_args=function_args,
-                function_name=function_name,
-            )
-            if is_break:
-                break
+        try:
+            async for event in response:
+                (
+                    is_break,
+                    has_function,
+                    function_name,
+                    function_args,
+                    completion,
+                ) = self._process_stream_event(
+                    event,
+                    chat=chat,
+                    has_function=has_function,
+                    completion=completion,
+                    function_args=function_args,
+                    function_name=function_name,
+                )
+                if is_break:
+                    break
+        except Exception:
+            pass
 
         print("")
         # TODO- get usage info in stream mode (?)

{langroid-0.1.240 → langroid-0.1.241}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langroid"
-version = "0.1.240"
+version = "0.1.241"
 description = "Harness LLMs with Multi-Agent Programming"
 authors = ["Prasad Chalasani <pchalasani@gmail.com>"]
 readme = "README.md"
@@ -100,6 +100,7 @@ groq = "^0.5.0"
 nest-asyncio = "^1.6.0"
 python-magic = "^0.4.27"
 transformers = {version="^4.40.1", optional=true}
+google-generativeai = "^0.5.2"
 
 [tool.poetry.extras]
 # install these using `poetry install -E [...]` where [...] is one of the extras below