Bump litellm and openai versions (#920)

Signed-off-by: Mandana Vaziri <[email protected]>

Author: vazirim

examples/rag/rag.py

@@ -105,7 +105,7 @@ def rag_retrieve(
         # Typically litellm.exceptions.APIConnectionError
         return f"Unexpected {type(be)}: be={be}"
 
-    data = response.data[0]["embedding"]
+    data = response.data[0]["embedding"]  # type: ignore
 
     milvus_client = get_or_create_client(database_name)
     search_res = milvus_client.search(

examples/rag/rag_library1.pdl

@@ -10,8 +10,8 @@ text:
   return:
     lang: python
     code: |
-        from examples.rag.rag import parse
-        result = parse(filename, chunk_size, chunk_overlap)
+        import rag
+        result = rag.parse(filename, chunk_size, chunk_overlap)
 - def: rag_index
   function:
     inp: list # This is a list[str], but PDL doesn't allow that type
@@ -22,8 +22,8 @@ text:
   return:
     lang: python
     code: |
-        from examples.rag.rag import rag_index
-        result = rag_index(inp, encoder_model, embed_dimension, database_name, collection_name)
+        import rag
+        result = rag.rag_index(inp, encoder_model, embed_dimension, database_name, collection_name)
 - def: rag_retrieve
   function:
     inp: str
@@ -34,5 +34,5 @@ text:
   return:
     lang: python
     code: |
-        from examples.rag.rag import rag_retrieve
-        result = rag_retrieve(inp, encoder_model, limit, database_name, collection_name)
+        import rag
+        result = rag.rag_retrieve(inp, encoder_model, limit, database_name, collection_name)

pyproject.toml

@@ -12,8 +12,8 @@ dependencies = [
   "jinja2~=3.0",
   "PyYAML~=6.0",
   "jsonschema~=4.0",
-  "litellm>=1.57.3,!=1.59.9,!=1.63.14",
-  "openai==1.61.0",
+  "litellm>=1.66.0",
+  "openai>=1.66.1",
   "termcolor~=2.0",
   "ipython>=8,<10",
   "json-repair~=0.35",

tests/test_ast_utils.py

@@ -25,11 +25,11 @@ class MapCounter:
     def __init__(self):
         self.cpt = 0
 
-    def count(map_self, ast):  # pylint: disable=no-self-argument
+    def count(map_self, ast):  # pylint: disable=no-self-argument # type: ignore
         map_self.cpt += 1
 
         class C(MappedFunctions):
-            def f_block(_, block):  # pylint: disable=no-self-argument
+            def f_block(_, block):  # pylint: disable=no-self-argument # type: ignore
                 return map_self.count(block)
 
         _ = map_block_children(C(), ast)

tests/test_examples_run.py

@@ -46,6 +46,11 @@
         pathlib.Path("pdl-live-react") / "demos" / "error.pdl",
         pathlib.Path("pdl-live-react") / "demos" / "demo1.pdl",
         pathlib.Path("pdl-live-react") / "demos" / "demo2.pdl",
+        pathlib.Path("pdl-live-react")
+        / "src-tauri"
+        / "tests"
+        / "cli"
+        / "read-stdin.pdl",
         # For now, skip the granite-io examples
         pathlib.Path("examples") / "granite-io" / "granite_io_hallucinations.pdl",
         pathlib.Path("examples") / "granite-io" / "granite_io_openai.pdl",