FAQ

from agency_swarm import set_openai_key
set_openai_key("YOUR_API_KEY")

OPENAI_API_KEY=sk-1234...

from dotenv import load_dotenv
load_dotenv()

import os
import json

def load_threads(chat_id: str) -> list[dict[str, TResponseInputItem]]:
    """Load all threads data for a specific chat session."""
    if os.path.exists(f"{chat_id}_threads.json"):
        with open(f"{chat_id}_threads.json", "r") as file:
            return json.load(file)
    return []

def save_threads(thread_dict: list[dict[str, TResponseInputItem]], chat_id: str):
    """Save all threads data to file."""
    with open(f"{chat_id}_threads.json", "w") as file:
        json.dump(thread_dict, file)

# Then, pass these callbacks during your agency initialization to resume conversations:
from agency_swarm import Agency
agency = Agency(
    agent,
    load_threads_callback=lambda: load_threads(chat_id),
    save_threads_callback=lambda thread_dict: save_threads(thread_dict, chat_id),
)

from fastapi import FastAPI
from uuid import uuid4

app = FastAPI()

@app.post("/chat")
async def chat(user_request: UserRequest):
    chat_id = user_request.chat_id or str(uuid4())

    agency = Agency(
        agent,
        load_threads_callback=lambda: load_threads(chat_id),
        save_threads_callback=lambda thread_dict: save_threads(thread_dict, chat_id)
    )

    response = await agency.get_response(user_request.message)
    return {"chat_id": chat_id, "response": response.final_output}

# Or use the built-in FastAPI integration
agency.run_fastapi(host="0.0.0.0", port=8000)