(20241209) SMS auth and sending ready.

models/behaviour/ai/llm/open_ai.py

@@ -45,6 +45,7 @@ from models.behaviour.base import BaseModel
 
 # Data Models:
 from models.data.api.ai.llm import LLMInput, LLMOutput, LLMUsageTokens
+from models.data.core.user_info import CoreUserInfoModel
 
 # To work with LLMs:
 from langchain_openai import ChatOpenAI
@@ -140,7 +141,7 @@ class LLMOpenAI(BaseModel):
     async def invoke(
             self,
             mongo_conn: AsyncMongo,
-            user_info: dict,
+            user_info: CoreUserInfoModel,
             llm_input: LLMInput
     ) -> LLMOutput:
 
@@ -167,12 +168,13 @@ class LLMOpenAI(BaseModel):
         )
 
         # Store this into MongoDB:
-        mongo_document = {"user": user_info}
+        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(
             collection = self.AI_USAGE_COLLECTION,
             document = mongo_document
         )
+        if inserted_id: llm_response.invocationId = str(inserted_id)
 
         # Done here:
         return llm_response