38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
import uuid
|
|
|
|
from fastapi import APIRouter, Depends, Query, HTTPException
|
|
from pydantic import BaseModel, Field
|
|
|
|
from memgpt.server.rest_api.interface import QueuingInterface
|
|
from memgpt.server.server import SyncServer
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
class ConfigRequest(BaseModel):
|
|
user_id: str = Field(..., description="Unique identifier of the user requesting the config.")
|
|
|
|
|
|
class ConfigResponse(BaseModel):
|
|
config: dict = Field(..., description="The server configuration object.")
|
|
|
|
|
|
def setup_config_index_router(server: SyncServer, interface: QueuingInterface):
|
|
@router.get("/config", tags=["config"], response_model=ConfigResponse)
|
|
def get_server_config(user_id: str = Query(..., description="Unique identifier of the user requesting the config.")):
|
|
"""
|
|
Retrieve the base configuration for the server.
|
|
"""
|
|
request = ConfigRequest(user_id=user_id)
|
|
|
|
# TODO remove once chatui adds user selection / pulls user from config
|
|
request.user_id = None if request.user_id == "null" else request.user_id
|
|
|
|
user_id = uuid.UUID(request.user_id) if request.user_id else None
|
|
|
|
interface.clear()
|
|
response = server.get_server_config(user_id=user_id)
|
|
return ConfigResponse(config=response)
|
|
|
|
return router
|