mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-10-26 21:26:39 +00:00
3182ecb1e4
* feat(subagents): preserve RAG evidence in durable batch exports * docs(subagents): clarify evidence omissions and migration head * docs(persistence): update the migration chain head * test(persistence): track the new batch evidence migration head * test(persistence): verify batch evidence migration on PostgreSQL * docs(subagents): retain evidence migration compatibility guidance * fix(subagents): propagate cited evidence forwarding omissions * fix(tests): adapt batch evidence PostgreSQL CI connection options * fix(tests): preserve TLS policy in batch evidence migration fixture
16 lines
586 B
Python
16 lines
586 B
Python
"""Adapt libpq test URIs to SQLAlchemy's asyncpg connection options."""
|
|
|
|
from sqlalchemy.engine import make_url
|
|
|
|
|
|
def asyncpg_test_url(uri: str) -> str:
|
|
url = make_url(uri)
|
|
query = dict(url.query)
|
|
if "sslmode" in query:
|
|
sslmode = query.pop("sslmode")
|
|
if "ssl" in query and query["ssl"] != sslmode:
|
|
raise ValueError("Conflicting ssl and sslmode in the test PostgreSQL URI")
|
|
query["ssl"] = sslmode
|
|
query.pop("channel_binding", None)
|
|
return url.set(drivername="postgresql+asyncpg", query=query).render_as_string(hide_password=False)
|