(20250116) Day-end push.

2025-01-16 19:10:50 +05:30
parent e5de9dace1
commit 75a71e2099
3 changed files with 290 additions and 6 deletions
@@ -128,7 +128,7 @@ class CoreLLMController(BaseModel):

    async def invoke(
            self,
-            mongo_conn: AsyncMongo,
+            mongo_data_conn: AsyncMongo,
            user_info: CoreUserInfoModel,
            llm_input: LLMInput
    ) -> LLMOutput:
@@ -158,7 +158,7 @@ class CoreLLMController(BaseModel):
        # Store this into MongoDB:
        mongo_document = {"user": user_info.model_dump()}
        for k, v in llm_response.model_dump().items(): mongo_document[k] = v
-        inserted_id = await mongo_conn.insert_one(
+        inserted_id = await mongo_data_conn.insert_one(
            collection = self.AI_USAGE_COLLECTION,
            document = mongo_document
        )