Fix various issues and refactor code (#1671)

* Update setup_mode to "Sync" in AstraDBVectorStoreComponent

* Fix search_kwargs default value in PineconeSearch and QdrantSearch components

* Update Playwright test and ChatMessage component in chat_io.spec.ts and chatMessage/index.tsx

* Refactor test_pickle_each_vertex function in test_graph.py

* Refactor database service to improve performance and readability
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-04-10 22:40:21 -03:00 • committed by GitHub
commit 4ee9b72634
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 451 additions and 83 deletions

View file

@ -416,17 +416,3 @@ async def test_pickle_graph(json_vector_store):
assert pickled is not None
unpickled = pickle.loads(pickled)
assert unpickled is not None
@pytest.mark.asyncio
async def test_pickle_each_vertex(json_vector_store):
starter_projects = load_starter_projects()
data = starter_projects[0][1]["data"]
graph = Graph.from_payload(data)
assert isinstance(graph, Graph)
for vertex in graph.vertices:
await vertex.build()
pickled = pickle.dumps(vertex)
assert pickled is not None
unpickled = pickle.loads(pickled)
assert unpickled is not None