Fix type declaration for the cross_encoder_model state variable. Update name of the new update API

This commit is contained in:
sabaimran 2023-12-21 09:15:13 +05:30
parent 089e4bee12
commit a1211f40d7
3 changed files with 10 additions and 2 deletions

View file

@ -334,7 +334,7 @@ async def update_chat_model(
@api.post("/config/data/search/model", status_code=200)
@requires(["authenticated"])
async def update_chat_model(
async def update_search_model(
request: Request,
id: str,
client: Optional[str] = None,

View file

@ -143,6 +143,13 @@ async def update(
)
return Response(content="Failed", status_code=500)
indexing_metadata = {
"num_org": len(org_files),
"num_markdown": len(markdown_files),
"num_pdf": len(pdf_files),
"num_plaintext": len(plaintext_files),
}
update_telemetry_state(
request=request,
telemetry_type="api",
@ -151,6 +158,7 @@ async def update(
user_agent=user_agent,
referer=referer,
host=host,
metadata=indexing_metadata,
)
logger.info(f"📪 Content index updated via API call by {client} client")

View file

@ -20,7 +20,7 @@ from khoj.utils.rawconfig import FullConfig
config = FullConfig()
search_models = SearchModels()
embeddings_model: Dict[str, EmbeddingsModel] = None
cross_encoder_model: CrossEncoderModel = None
cross_encoder_model: Dict[str, CrossEncoderModel] = None
content_index = ContentIndex()
openai_client: OpenAI = None
gpt4all_processor_config: GPT4AllProcessorModel = None