diff --git a/README.md b/README.md index 9bb01f5..e9d4b5a 100644 --- a/README.md +++ b/README.md @@ -52,6 +52,7 @@ from openbb_ai import ( app = FastAPI() + @app.get("/agents.json") async def agents_json(): return JSONResponse( @@ -60,7 +61,9 @@ async def agents_json(): "name": "My Agent", "description": "This is my agent", "image": f"{AGENT_BASE_URL}/my-agent/logo.png", - "endpoints": {"query": f"{AGENT_BASE_URL}/query"}, # must match the query endpoint below + "endpoints": { + "query": f"{AGENT_BASE_URL}/query" + }, # must match the query endpoint below "features": { "streaming": True, # must be True "widget-dashboard-select": True, # Enable access to priority widgets @@ -70,6 +73,7 @@ async def agents_json(): } ) + @app.get("/query") async def stream(request: QueryRequest): async def execution_loop(): diff --git a/openbb_ai/models.py b/openbb_ai/models.py index e365d32..ffb38bf 100644 --- a/openbb_ai/models.py +++ b/openbb_ai/models.py @@ -908,6 +908,7 @@ class FunctionCallSSEData(BaseModel): "execute_agent_tool", "manage_navigation_bar", "get_skill_content", + "save_skill", ] input_arguments: dict extra_state: dict | None = Field( diff --git a/pyproject.toml b/pyproject.toml index 4a21d39..0fd28bd 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "openbb-ai" -version = "2.1.0" +version = "2.2.0" description = "An SDK for building agents compatible with OpenBB Workspace" authors = [ { name = "OpenBB Team", email = "hello@openbb.finance" },