ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-13 16:53:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: update text and text_ids value is ""

Browse Source
This commit is contained in:
ACBBZ
2025-08-19 16:46:32 +08:00
parent e55d5c0a10
commit ced10e4447
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/blackbox/chroma_upsert.py
View File

@ -122,7 +122,7 @@ class ChromaUpsert(Blackbox):
response_file = f"\n{file} ids is 0~{len(docs)-1}" response_file = f"\n{file} ids is 0~{len(docs)-1}"
if text is not None and text_ids is not None: if text != "" and text_ids != "":
Chroma.from_texts(texts=[text], embedding=embedding_model, ids=[text_ids], collection_name=chroma_collection_id, client=client) Chroma.from_texts(texts=[text], embedding=embedding_model, ids=[text_ids], collection_name=chroma_collection_id, client=client)
response_string = f"\n{text} ids is {text_ids}" response_string = f"\n{text} ids is {text_ids}"
@ -140,8 +140,8 @@ class ChromaUpsert(Blackbox):
async def fast_api_handler(self, request: Request) -> Response: async def fast_api_handler(self, request: Request) -> Response:
user_file = (await request.form()).get("file") user_file = (await request.form()).get("file")
user_text = (await request.form()).get("text") user_text = (await request.form()).get("text", "")
user_text_ids = (await request.form()).get("text_ids") user_text_ids = (await request.form()).get("text_ids", "")
setting: dict = (await request.form()).get("settings") setting: dict = (await request.form()).get("settings")
if isinstance(setting, str): if isinstance(setting, str):
@ -150,13 +150,13 @@ class ChromaUpsert(Blackbox):
except json.JSONDecodeError: except json.JSONDecodeError:
return JSONResponse(content={"error": "Invalid settings format"}, status_code=status.HTTP_400_BAD_REQUEST) return JSONResponse(content={"error": "Invalid settings format"}, status_code=status.HTTP_400_BAD_REQUEST)
if user_file is None and user_text is None: if user_file is None and user_text == "":
return JSONResponse(content={"error": "file or text is required"}, status_code=status.HTTP_400_BAD_REQUEST) return JSONResponse(content={"error": "file or text is required"}, status_code=status.HTTP_400_BAD_REQUEST)
if user_text is not None and user_text_ids is None: if user_text != "" and user_text_ids == "":
return JSONResponse(content={"error": "text_ids is required when text is provided"}, status_code=status.HTTP_400_BAD_REQUEST) return JSONResponse(content={"error": "text_ids is required when text is provided"}, status_code=status.HTTP_400_BAD_REQUEST)
if user_file is not None and user_file.size != 0: if user_file != None and user_file.size != 0:
pdf_bytes = await user_file.read() pdf_bytes = await user_file.read()
custom_filename = user_file.filename custom_filename = user_file.filename