update

main.py

@@ -7,7 +7,7 @@ import tiktoken
 from uuid import uuid4
 from functools import lru_cache
 from typing import Optional, List, Dict, Literal
-from fastapi import FastAPI, HTTPException, Depends, Security, BackgroundTasks
+from fastapi import FastAPI, HTTPException, Depends, Security, BackgroundTasks, Query
 from fastapi.security import APIKeyHeader
 from fastapi.responses import StreamingResponse
 from pydantic import BaseModel, Field