Fixing tests

ai-forever · Oct 7, 2024 · 006c350 · 006c350
1 parent cfc1908
commit 006c350
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/libs/community/tests/integration_tests/adapters/test_openai.py b/libs/community/tests/integration_tests/adapters/test_openai.py
@@ -6,7 +6,7 @@
 def _test_no_stream(**kwargs: Any) -> None:
     import openai
 
-    result = openai.ChatCompletion.create(**kwargs)
+    result = openai.ChatCompletion.create(**kwargs)  # type: ignore
     lc_result = lcopenai.ChatCompletion.create(**kwargs)
     if isinstance(lc_result, dict):
         if isinstance(result, dict):
@@ -20,7 +20,7 @@ def _test_stream(**kwargs: Any) -> None:
     import openai
 
     result = []
-    for c in openai.ChatCompletion.create(**kwargs):
+    for c in openai.ChatCompletion.create(**kwargs):  # type: ignore
         result.append(c["choices"][0]["delta"].to_dict_recursive())
 
     lc_result = []
@@ -32,7 +32,7 @@ def _test_stream(**kwargs: Any) -> None:
 async def _test_async(**kwargs: Any) -> None:
     import openai
 
-    result = await openai.ChatCompletion.acreate(**kwargs)
+    result = await openai.ChatCompletion.acreate(**kwargs)  # type: ignore
     lc_result = await lcopenai.ChatCompletion.acreate(**kwargs)
     if isinstance(lc_result, dict):
         if isinstance(result, dict):
@@ -46,7 +46,7 @@ async def _test_astream(**kwargs: Any) -> None:
     import openai
 
     result = []
-    async for c in await openai.ChatCompletion.acreate(**kwargs):
+    async for c in await openai.ChatCompletion.acreate(**kwargs):  # type: ignore
         result.append(c["choices"][0]["delta"].to_dict_recursive())
 
     lc_result = []

diff --git a/libs/community/tests/integration_tests/chains/test_ontotext_graphdb_qa.py b/libs/community/tests/integration_tests/chains/test_ontotext_graphdb_qa.py
@@ -375,7 +375,7 @@ def test_chain(model_name: str, question: str) -> None:
         "FROM <https://swapi.co/ontology/> WHERE {?s ?p ?o}",
     )
     chain = OntotextGraphDBQAChain.from_llm(
-        ChatOpenAI(temperature=0, model_name=model_name),
+        ChatOpenAI(temperature=0, model=model_name),
         graph=graph,
         verbose=True,  # type: ignore[call-arg]
     )

diff --git a/libs/community/tests/integration_tests/embeddings/test_azure_openai.py b/libs/community/tests/integration_tests/embeddings/test_azure_openai.py
@@ -105,7 +105,7 @@ def test_azure_openai_embedding_with_empty_string() -> None:
     output = embedding.embed_documents(document)
     assert len(output) == 2
     assert len(output[0]) == 1536
-    expected_output = openai.Embedding.create(input="", model="text-embedding-ada-002")[
+    expected_output = openai.Embedding.create(input="", model="text-embedding-ada-002")[  # type: ignore
         "data"
     ][0]["embedding"]
     assert np.allclose(output[0], expected_output)

diff --git a/libs/community/tests/integration_tests/embeddings/test_openai.py b/libs/community/tests/integration_tests/embeddings/test_openai.py
@@ -71,7 +71,7 @@ def test_openai_embedding_with_empty_string() -> None:
     output = embedding.embed_documents(document)
     assert len(output) == 2
     assert len(output[0]) == 1536
-    expected_output = openai.Embedding.create(input="", model="text-embedding-ada-002")[
+    expected_output = openai.Embedding.create(input="", model="text-embedding-ada-002")[  # type: ignore
         "data"
     ][0]["embedding"]
     assert np.allclose(output[0], expected_output)