langroid 0.1.240__tar.gz → 0.1.243__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langroid
-Version: 0.1.240
+Version: 0.1.243
 Summary: Harness LLMs with Multi-Agent Programming
 License: MIT
 Author: Prasad Chalasani
@@ -30,12 +30,13 @@ Requires-Dist: chainlit (>=1.0.400,<2.0.0) ; extra == "chainlit"
 Requires-Dist: chromadb (>=0.4.21,<=0.4.23) ; extra == "chromadb"
 Requires-Dist: colorlog (>=6.7.0,<7.0.0)
 Requires-Dist: docstring-parser (>=0.15,<0.16)
-Requires-Dist: duckduckgo-search (>=5.3.0,<6.0.0)
+Requires-Dist: duckduckgo-search (>=6.0.0,<7.0.0)
 Requires-Dist: faker (>=18.9.0,<19.0.0)
 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.243}/langroid/language_models/openai_gpt.py

@@ -71,6 +71,7 @@ class OpenAIChatModel(str, Enum):
     GPT4 = "gpt-4"
     GPT4_32K = "gpt-4-32k"
     GPT4_TURBO = "gpt-4-turbo"
+    GPT4o = "gpt-4o"
 
 
 class OpenAICompletionModel(str, Enum):
@@ -86,6 +87,7 @@ _context_length: Dict[str, int] = {
     OpenAIChatModel.GPT4: 8192,
     OpenAIChatModel.GPT4_32K: 32_768,
     OpenAIChatModel.GPT4_TURBO: 128_000,
+    OpenAIChatModel.GPT4o: 128_000,
     OpenAICompletionModel.TEXT_DA_VINCI_003: 4096,
 }
 
@@ -95,10 +97,12 @@ _cost_per_1k_tokens: Dict[str, Tuple[float, float]] = {
     OpenAIChatModel.GPT3_5_TURBO: (0.001, 0.002),
     OpenAIChatModel.GPT4: (0.03, 0.06),  # 8K context
     OpenAIChatModel.GPT4_TURBO: (0.01, 0.03),  # 128K context
+    OpenAIChatModel.GPT4o: (0.005, 0.015),  # 128K context
 }
 
 
 openAIChatModelPreferenceList = [
+    OpenAIChatModel.GPT4o,
     OpenAIChatModel.GPT4_TURBO,
     OpenAIChatModel.GPT4,
     OpenAIChatModel.GPT3_5_TURBO,
@@ -670,23 +674,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 +729,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.243}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langroid"
-version = "0.1.240"
+version = "0.1.243"
 description = "Harness LLMs with Multi-Agent Programming"
 authors = ["Prasad Chalasani <pchalasani@gmail.com>"]
 readme = "README.md"
@@ -90,7 +90,7 @@ grpcio = "^1.62.1"
 metaphor-python = {version = "^0.1.23", optional = true}
 chainlit = {version = "^1.0.400", optional = true}
 python-socketio = {version="^5.11.0", optional=true}
-duckduckgo-search = "^5.3.0"
+duckduckgo-search = "^6.0.0"
 huggingface-hub = {version="^0.21.2", optional=true}
 pdf2image = "^1.17.0"
 pytesseract = "^0.3.10"
@@ -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