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

update chroma chat

Browse Source
This commit is contained in:
ACBBZ
2024-05-07 06:24:35 +00:00
parent a6b2d6c3e7
commit 31dbe29b8b
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/blackbox/chroma_chat.py
View File

@ -4,6 +4,9 @@ from fastapi import Request, Response, status
from fastapi.responses import JSONResponse from fastapi.responses import JSONResponse
from .blackbox import Blackbox from .blackbox import Blackbox
from ..utils import chroma_setting
DEFAULT_COLLECTION_ID = "123"
from injector import singleton from injector import singleton
@singleton @singleton
class ChromaChat(Blackbox): class ChromaChat(Blackbox):
@ -19,13 +22,13 @@ class ChromaChat(Blackbox):
data = args[0] data = args[0]
return isinstance(data, list) return isinstance(data, list)
def processing(self, question, context) -> str: def processing(self, question, setting: chroma_setting) -> str:
# load or create collection # load or create collection
if context is None: if setting is None:
collection_id = "123" collection_id = DEFAULT_COLLECTION_ID
else: else:
collection_id = context["collections"][0] collection_id = setting.ChromaSetting.collection_ids[0]
# query it # query it
chroma_result = self.chroma_query(question, collection_id) chroma_result = self.chroma_query(question, collection_id)