55from fastapi import FastAPI
66import requests
77from sqlmodel import Session , select
8- from apps .chat .task .llm import LLMService
98from apps .datasource .models .datasource import CoreDatasource
109from apps .system .models .system_model import AssistantModel
1110from apps .system .schemas .auth import CacheName , CacheNamespace
1413from common .core .db import engine
1514from starlette .middleware .cors import CORSMiddleware
1615from common .core .config import settings
17- from deps import CurrentUser
1816
1917@cache (namespace = CacheNamespace .EMBEDDED_INFO , cacheName = CacheName .ASSISTANT_INFO , keyExpression = "assistant_id" )
2018async def get_assistant_info (* , session : Session , assistant_id : int ) -> AssistantModel | None :
@@ -25,7 +23,7 @@ def get_assistant_user(*, id: int):
2523 return UserInfoDTO (
id = id ,
account = "sqlbot-inner-assistant" ,
oid = 1 ,
name = "sqlbot-inner-assistant" ,
email = "[email protected] " )
2624
2725# def get_assistant_ds(*, session: Session, assistant: AssistantModel):
28- def get_assistant_ds (llm_service : LLMService ) -> list [dict ]:
26+ def get_assistant_ds (llm_service ) -> list [dict ]:
2927 assistant : AssistantModel = llm_service .current_assistant
3028 session : Session = llm_service .session
3129 type = assistant .type
0 commit comments