diff --git a/app/server/ChatbotAgent/chatbot_server.py b/app/server/ChatbotAgent/chatbot_server.py index 6c1d0d9..872006e 100644 --- a/app/server/ChatbotAgent/chatbot_server.py +++ b/app/server/ChatbotAgent/chatbot_server.py @@ -25,26 +25,26 @@ class PredictRequest(BaseModel): class LCChatBot(ls.LitAPI): def setup(self, device): - self.llm = AsyncGeminiLLM(model_name=settings.LLM_MODEL_NAME) + # self.llm = AsyncGeminiLLM(model_name=settings.LLM_MODEL_NAME) self.redis = RedisManager( host=settings.REDIS_HOST, port=settings.REDIS_PORT, db=settings.REDIS_DB, key_prefix=settings.REDIS_HISTORY_KEY_PREFIX ) - self.vector_db = VectorDatabase( - vector_db_dir=settings.VECTOR_DB_DIR, - collection_name=settings.COLLECTION_NAME, - embedding_model_name=settings.EMBEDDING_MODEL_NAME - ) - self.stylist_agent_kwages = { - 'local_db': self.vector_db, - 'max_len': 5, - 'outfits_root': settings.OUTFIT_OUTPUT_DIR, - 'image_dir': settings.IMAGE_DIR, - 'stylist_guide_dir': settings.STYLIST_GUIDE_DIR, - 'gemini_model_name': settings.LLM_MODEL_NAME - } + # self.vector_db = VectorDatabase( + # vector_db_dir=settings.VECTOR_DB_DIR, + # collection_name=settings.COLLECTION_NAME, + # embedding_model_name=settings.EMBEDDING_MODEL_NAME + # ) + # self.stylist_agent_kwages = { + # 'local_db': self.vector_db, + # 'max_len': 5, + # 'outfits_root': settings.OUTFIT_OUTPUT_DIR, + # 'image_dir': settings.IMAGE_DIR, + # 'stylist_guide_dir': settings.STYLIST_GUIDE_DIR, + # 'gemini_model_name': settings.LLM_MODEL_NAME + # } self.gemini_client = genai.Client( vertexai=True, project='aida-461108', location='us-central1' )