Fix testcase - agentframework v2 - Refactor tests for ChatWithDataPlugin, ChatService, HistoryService, and App

- Removed test cases for deprecated agent factories in test_app.py.
- Updated test cases in test_chat_with_data_plugin.py to reflect changes in the ChatWithDataPlugin implementation.
- Refactored test cases in test_chat_service.py to remove dependencies on mock requests and streamline the setup.
- Enhanced test_generate_title in test_history_service.py to utilize the new v2 agent framework.
- Improved error handling tests in test_chat_service.py for rate limits and general exceptions.
- Consolidated mock setups and assertions for clarity and maintainability.

Author: Pavan-Microsoft

src/api/common/database/sqldb_service.py

@@ -9,6 +9,7 @@
 from helpers.azure_credential_utils import get_azure_credential_async
 import pyodbc
 
+
 class SQLTool(BaseModel):
     model_config = {"arbitrary_types_allowed": True}
     conn: pyodbc.Connection
@@ -27,6 +28,7 @@ async def get_sql_response(self, sql_query: str) -> str:
             if cursor:
                 cursor.close()
 
+
 async def get_db_connection():
     """Get a connection to the SQL database"""
     config = Config()

src/api/services/chat_service.py

@@ -10,7 +10,6 @@
 import json
 import logging
 import re
-import os
 
 from helpers.azure_credential_utils import get_azure_credential_async
 from common.database.sqldb_service import SQLTool, get_db_connection as get_sqldb_connection
@@ -87,6 +86,7 @@ async def _delete_thread_async(self, thread_conversation_id: str):
             if credential is not None:
                 await credential.close()
 
+
 class ChatService:
     """
     Service for handling chat interactions, including streaming responses,
@@ -155,7 +155,7 @@ async def stream_openai_text(self, conversation_id: str, query: str) -> Streamin
                             for content in chunk.contents:
                                 if hasattr(content, "annotations") and content.annotations:
                                     citations.extend(content.annotations)
-                        
+
                         if first_chunk:
                             if chunk is not None and chunk.text != "":
                                 first_chunk = False
@@ -166,6 +166,7 @@ async def stream_openai_text(self, conversation_id: str, query: str) -> Streamin
 
                     if ChatService.thread_cache is not None and thread is not None:
                         ChatService.thread_cache[conversation_id] = thread_conversation_id
+
                     if citations:
                         citation_list = [f"{{\"url\": \"{citation.url}\", \"title\": \"{citation.title}\"}}" for citation in citations]
                         yield ", \"citations\": [" + ",".join(citation_list) + "]}"