wehub-resource-sync
60 lines
1.3 KiB
Python
60 lines
1.3 KiB
Python
"""Demo"""
|
|
|
|
import os
|
|
|
|
import uvicorn
|
|
from ag_ui_langgraph import add_langgraph_fastapi_endpoint
|
|
from copilotkit import LangGraphAGUIAgent
|
|
from dotenv import load_dotenv
|
|
from fastapi import FastAPI
|
|
|
|
load_dotenv()
|
|
os.environ["LANGGRAPH_FASTAPI"] = "true"
|
|
from src.agent import graph # noqa: E402
|
|
|
|
app = FastAPI()
|
|
|
|
add_langgraph_fastapi_endpoint(
|
|
app=app,
|
|
agent=LangGraphAGUIAgent(
|
|
name="research_agent", description="Research agent.", graph=graph
|
|
),
|
|
path="/copilotkit/agents/research_agent",
|
|
)
|
|
# add_langgraph_fastapi_endpoint(
|
|
# app=app,
|
|
# agent=LangGraphAGUIAgent(
|
|
# name="research_agent_google_genai", description="Research agent.", graph=graph
|
|
# ),
|
|
# path="/copilotkit/agents/research_agent_google_genai",
|
|
# )
|
|
|
|
|
|
@app.get("/health")
|
|
def health():
|
|
"""Health check."""
|
|
return {"status": "ok"}
|
|
|
|
|
|
def main():
|
|
"""Run the uvicorn server."""
|
|
port = int(os.getenv("PORT", "8000"))
|
|
uvicorn.run(
|
|
"main:app",
|
|
host="0.0.0.0",
|
|
port=port,
|
|
reload=True,
|
|
reload_dirs=(
|
|
["."]
|
|
+ (
|
|
["../../../../sdk-python/copilotkit"]
|
|
if os.path.exists("../../../../sdk-python/copilotkit")
|
|
else []
|
|
)
|
|
),
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|