cod

Author: howieleung

.github/workflows/azure-dev.yml

@@ -1,11 +1,11 @@
-# Run when commits are pushed to main
+# Run when commits are pushed to howie/load-chat-history
 on:
   workflow_dispatch:
   push:
     # Run when commits are pushed to mainline branch (main or master)
     # Set this to the mainline branch you are using
     branches:
-      - main
+      - howie/load-chat-history
 
 # Set up permissions for deploying with secretless Azure federated credentials
 # https://learn.microsoft.com/en-us/azure/developer/github/connect-from-azure?tabs=azure-portal%2Clinux#set-up-azure-login-with-openid-connect-authentication
@@ -50,6 +50,8 @@ jobs:
       AZURE_AI_EMBED_MODEL_NAME: ${{ vars.AZURE_AI_EMBED_MODEL_NAME }}
       AZURE_AI_EMBED_MODEL_FORMAT: ${{ vars.AZURE_AI_EMBED_MODEL_FORMAT }}
       AZURE_AI_EMBED_MODEL_VERSION: ${{ vars.AZURE_AI_EMBED_MODEL_VERSION }}
+      AZURE_AI_EMBED_DIMENSIONS: ${{ vars.AZURE_AI_EMBED_DIMENSIONS }}
+      AZURE_AI_SEARCH_INDEX_NAME: ${{ vars.AZURE_AI_SEARCH_INDEX_NAME }}
       AZURE_EXISTING_AIPROJECT_CONNECTION_STRING: ${{ vars.AZURE_EXISTING_AIPROJECT_CONNECTION_STRING }}
     steps:
       - name: Checkout

.github/workflows/template-validation.yml

@@ -52,7 +52,7 @@ def get_agent(request: Request) -> Agent:
 def serialize_sse_event(data: Dict) -> str:
     return f"data: {json.dumps(data)}\n\n"
 
-async def get_completed_message(ai_client : AIProjectClient, message: ThreadMessage) -> Dict:
+async def get_message_and_annotations(ai_client : AIProjectClient, message: ThreadMessage) -> Dict:
         annotations = []
         # Get file annotations for the file search.
         for annotation in (a.as_dict() for a in message.file_citation_annotations):
@@ -72,8 +72,7 @@ async def get_completed_message(ai_client : AIProjectClient, message: ThreadMess
 
         return {
                 'content': message.text_messages[0].text.value,
-                'annotations': annotations,
-                'type': "completed_message"
+                'annotations': annotations
             }
 class MyEventHandler(AsyncAgentEventHandler[str]):
     def __init__(self, ai_client: AIProjectClient):
@@ -92,7 +91,8 @@ async def on_thread_message(self, message: ThreadMessage) -> Optional[str]:
 
             logger.info("MyEventHandler: Received completed message")
 
-            stream_data = await get_completed_message(self.ai_client, message)
+            stream_data = await get_message_and_annotations(self.ai_client, message)
+            stream_data['type'] = "completed_message"
             return serialize_sse_event(stream_data)
         except Exception as e:
             logger.error(f"Error in event handler for thread message: {e}", exc_info=True)
@@ -185,10 +185,11 @@ async def history(
     # Create a new message from the user's input.
     try:
         content = []
-        for message in await ai_client.agents.list_messages(
+        response = await ai_client.agents.list_messages(
             thread_id=thread_id,
-        ):
-            content.append(await get_completed_message(ai_client, message))
+        )
+        for message in response.data:
+            content.append(await get_message_and_annotations(ai_client, message))
 
         logger.info(f"List message, thread ID: {thread_id}")
         response = JSONResponse(content=content)