WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-08-26 09:32:25 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix uri parsing

Browse Source
This commit is contained in:
William Fu-Hinthorn
2024-11-22 08:24:50 -08:00
parent e5c0f9b5ab
commit 8eaebf6b6d
2 changed files with 16 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/checkpoint-postgres/tests/test_async_store.py
+8 -2
View File
@@ -17,8 +17,14 @@ async def store(request) -> AsyncIterator[AsyncPostgresStore]:
pytest.skip("Async Postgres tests require Python 3.10+")
database = f"test_{uuid.uuid4().hex[:16]}"
uri_base = DEFAULT_URI.rsplit("/", 2)[0]
conn_string = f"{uri_base}/{database}?sslmode=disable"
uri_parts = DEFAULT_URI.split("/")
uri_base = "/".join(uri_parts[:-1])
query_params = ""
if "?" in uri_parts[-1]:
db_name, query_params = uri_parts[-1].split("?", 1)
query_params = "?" + query_params
conn_string = f"{uri_base}/{database}{query_params}"
admin_conn_string = DEFAULT_URI
async with await AsyncConnection.connect(
libs/checkpoint-postgres/tests/test_store.py
+8 -4
View File
@@ -21,10 +21,14 @@ from langgraph.store.postgres import PostgresStore
@pytest.fixture(scope="function", params=["default", "pipe", "pool"])
def store(request) -> PostgresStore:
database = f"test_{uuid4().hex[:16]}"
uri_base = DEFAULT_URI.rsplit("/", 2)[
0
] # Get everything before /postgres?sslmode=disable
conn_string = f"{uri_base}/{database}?sslmode=disable"
uri_parts = DEFAULT_URI.split("/")
uri_base = "/".join(uri_parts[:-1])
query_params = ""
if "?" in uri_parts[-1]:
db_name, query_params = uri_parts[-1].split("?", 1)
query_params = "?" + query_params
conn_string = f"{uri_base}/{database}{query_params}"
admin_conn_string = DEFAULT_URI
with Connection.connect(admin_conn_string, autocommit=True) as conn: