WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-28 20:45:05 +02:00
Code Issues Packages Projects Releases Wiki Activity

merge

Browse Source
This commit is contained in:
William Fu-Hinthorn
2024-11-27 14:13:43 -08:00
parent 2bff330bd7
commit d3b2e2dc95
5 changed files with 94 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/checkpoint-postgres/tests/test_async_store.py
+6 -6
View File
@@ -212,7 +212,7 @@ async def _create_vector_store(
conn_string = f"{uri_base}/{database}{query_params}"
admin_conn_string = DEFAULT_URI
embedding_config = {
index_config = {
"dims": fake_embeddings.dims,
"embed": fake_embeddings,
"index_config": {
@@ -230,7 +230,7 @@ async def _create_vector_store(
try:
async with AsyncPostgresStore.from_conn_string(
conn_string,
embedding=embedding_config,
embedding=index_config,
) as store:
await store.setup()
yield store
@@ -271,10 +271,10 @@ async def test_vector_store_initialization(
vector_store: AsyncPostgresStore, fake_embeddings: CharacterEmbeddings
) -> None:
"""Test store initialization with embedding config."""
assert vector_store.embedding_config is not None
assert vector_store.embedding_config["dims"] == fake_embeddings.dims
if isinstance(vector_store.embedding_config["embed"], Embeddings):
assert vector_store.embedding_config["embed"] == fake_embeddings
assert vector_store.index_config is not None
assert vector_store.index_config["dims"] == fake_embeddings.dims
if isinstance(vector_store.index_config["embed"], Embeddings):
assert vector_store.index_config["embed"] == fake_embeddings
async def test_vector_insert_with_auto_embedding(
libs/checkpoint-postgres/tests/test_store.py
+5 -5
View File
@@ -370,7 +370,7 @@ def _create_vector_store(
conn_string = f"{uri_base}/{database}{query_params}"
admin_conn_string = DEFAULT_URI
embedding_config = {
index_config = {
"dims": fake_embeddings.dims,
"embed": fake_embeddings,
"index_config": {
@@ -386,7 +386,7 @@ def _create_vector_store(
try:
with PostgresStore.from_conn_string(
conn_string,
embedding=embedding_config,
embedding=index_config,
) as store:
store.setup()
yield store
@@ -426,9 +426,9 @@ def test_vector_store_initialization(
) -> None:
"""Test store initialization with embedding config."""
# Store should be initialized with embedding config
assert vector_store.embedding_config is not None
assert vector_store.embedding_config["dims"] == fake_embeddings.dims
assert vector_store.embedding_config["embed"] == fake_embeddings
assert vector_store.index_config is not None
assert vector_store.index_config["dims"] == fake_embeddings.dims
assert vector_store.index_config["embed"] == fake_embeddings
def test_vector_insert_with_auto_embedding(vector_store: PostgresStore) -> None: