Skip to content

Commit 74dc895

Browse files
committed
fix: add configurable PORT env var to AGUI templates + update snapshots
All 3 AGUI templates now read PORT from env with default 8080: uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", "8080"))) Addresses PR review comment requesting configurable port for local testing.
1 parent 4a168ed commit 74dc895

4 files changed

Lines changed: 8 additions & 7 deletions

File tree

src/assets/__tests__/__snapshots__/assets.snapshot.test.ts.snap

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1849,7 +1849,7 @@ async def ping():
18491849
18501850
18511851
if __name__ == "__main__":
1852-
uvicorn.run(app, host="0.0.0.0", port=8080)
1852+
uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", "8080")))
18531853
"
18541854
`;
18551855
@@ -2012,6 +2012,7 @@ from fastapi import FastAPI
20122012
from fastapi.middleware.cors import CORSMiddleware
20132013
from langgraph.graph import StateGraph, START
20142014
from langgraph.graph.message import MessagesState
2015+
from langgraph.checkpoint.memory import MemorySaver
20152016
from langgraph.prebuilt import ToolNode, tools_condition
20162017
from langchain_core.tools import tool
20172018
from opentelemetry.instrumentation.langchain import LangchainInstrumentor
@@ -2049,7 +2050,7 @@ builder.add_node("tools", ToolNode(tools=backend_tools))
20492050
builder.add_edge(START, "chat")
20502051
builder.add_conditional_edges("chat", tools_condition)
20512052
builder.add_edge("tools", "chat")
2052-
graph = builder.compile()
2053+
graph = builder.compile(checkpointer=MemorySaver())
20532054
20542055
agent = LangGraphAgent(
20552056
name="{{ name }}",
@@ -2074,7 +2075,7 @@ async def ping():
20742075
20752076
20762077
if __name__ == "__main__":
2077-
uvicorn.run(app, host="0.0.0.0", port=8080)
2078+
uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", "8080")))
20782079
"
20792080
`;
20802081
@@ -2358,7 +2359,7 @@ agui_agent = StrandsAgent(agent=agent, name="{{ name }}", description="A helpful
23582359
app = create_strands_app(agui_agent, path="/invocations", ping_path="/ping")
23592360
23602361
if __name__ == "__main__":
2361-
uvicorn.run(app, host="0.0.0.0", port=8080)
2362+
uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", "8080")))
23622363
"
23632364
`;
23642365

src/assets/python/agui/googleadk/base/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,4 +24,4 @@ async def ping():
2424

2525

2626
if __name__ == "__main__":
27-
uvicorn.run(app, host="0.0.0.0", port=8080)
27+
uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", "8080")))

src/assets/python/agui/langchain_langgraph/base/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,4 +71,4 @@ async def ping():
7171

7272

7373
if __name__ == "__main__":
74-
uvicorn.run(app, host="0.0.0.0", port=8080)
74+
uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", "8080")))

src/assets/python/agui/strands/base/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,4 +40,4 @@ def session_manager_provider(input_data):
4040
app = create_strands_app(agui_agent, path="/invocations", ping_path="/ping")
4141

4242
if __name__ == "__main__":
43-
uvicorn.run(app, host="0.0.0.0", port=8080)
43+
uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", "8080")))

0 commit comments

Comments
 (0)