mirror of
https://github.com/microsoft/autogen.git
synced 2025-10-31 01:40:58 +00:00
31 lines
865 B
Python
31 lines
865 B
Python
# loads a fast api api endpoint with a single endpoint that takes text query and return a response
|
|||
|
|||
import json
|
|||
import os
|
|||
|
|||
from fastapi import FastAPI
|
|||
|
|||
from ..datamodel import Response
|
|||
from ..workflowmanager import WorkflowManager
|
|||
|
|||
app = FastAPI()
|
|||
workflow_file_path = os.environ.get("AUTOGENSTUDIO_WORKFLOW_FILE", None)
|
|||
|
|||
|
|||
if workflow_file_path:
|
|||
workflow_manager = WorkflowManager(workflow=workflow_file_path)
|
|||
else:
|
|||
raise ValueError("Workflow file must be specified")
|
|||
|
|||
|
|||
@app.get("/predict/{task}")
|
|||
async def predict(task: str):
|
|||
response = Response(message="Task successfully completed", status=True, data=None)
|
|||
try:
|
|||
result_message = workflow_manager.run(message=task, clear_history=False)
|
|||
response.data = result_message
|
|||
except Exception as e:
|
|||
response.message = str(e)
|
|||
response.status = False
|
|||
return response
|
