generated from kgod/ai-review-template
feat: initialize resume agent with OfferPai sync
This commit is contained in:
@@ -0,0 +1,29 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
|
||||
from sqlalchemy import create_engine, text
|
||||
|
||||
|
||||
def test_langgraph_runtime_is_available() -> None:
|
||||
from langgraph.graph import START, StateGraph
|
||||
|
||||
graph = StateGraph(dict)
|
||||
graph.add_node("finish", lambda state: state)
|
||||
graph.add_edge(START, "finish")
|
||||
|
||||
assert graph.compile().invoke({}) == {}
|
||||
|
||||
|
||||
def test_postgres_test_database_has_pgvector() -> None:
|
||||
database_url = os.environ["RESUME_AGENT_TEST_DATABASE_URL"]
|
||||
engine = create_engine(database_url)
|
||||
try:
|
||||
with engine.connect() as connection:
|
||||
extension = connection.execute(
|
||||
text("SELECT extname FROM pg_extension WHERE extname = 'vector'")
|
||||
).scalar_one_or_none()
|
||||
finally:
|
||||
engine.dispose()
|
||||
|
||||
assert extension == "vector"
|
||||
Reference in New Issue
Block a user