Azure openai fixes (#2428)

2025-03-25 00:34:21 +05:30
parent 2b49c9eedd
commit 953a5a4a2d
4 changed files with 17 additions and 6 deletions
--- a/tests/vector_stores/test_pinecone.py
+++ b/tests/vector_stores/test_pinecone.py
@@ -67,7 +67,7 @@ def test_insert_vectors(pinecone_db):

 def test_search_vectors(pinecone_db):
    pinecone_db.index.query.return_value.matches = [{"id": "id1", "score": 0.9, "metadata": {"name": "vector1"}}]
-    results = pinecone_db.search([0.1] * 128, limit=1)
+    results = pinecone_db.search("test query",[0.1] * 128, limit=1)
    assert len(results) == 1
    assert results[0].id == "id1"
    assert results[0].score == 0.9