testing changes in main not to be pushed, just push RR_collab_SK version

Author: Heena Ugale

FETCH_HEAD

@@ -0,0 +1,16 @@
+import requests
+
+# Backend URL (adjust if using a different port)
+url = "http://localhost:7000/chat"
+
+# Prepare your payload with session_id and user prompt
+data = {
+    "session_id": "test_session_1",
+    "prompt": "what was my last subscription for customer ID 103?"
+}
+
+# Make the POST request to the backend
+response = requests.post(url, json=data)
+
+# Print the response
+print("Response:", response.json())

agentic_ai/agents/semantic_kernel/multi_agent/backend_test.py

@@ -0,0 +1,41 @@
+import os  
+import logging  
+from typing import Any, Dict, List, Optional  
+from dotenv import load_dotenv  
+  
+load_dotenv()  # Load environment variables from .env file if needed  
+  
+class BaseAgent:  
+    """  
+    Base class for all agents.  
+    Not intended to be used directly.  
+    Handles environment variables, state store, and chat history.  
+    """  
+  
+    def __init__(self, state_store: Dict[str, Any], session_id: str) -> None:  
+        self.azure_deployment = os.getenv("AZURE_OPENAI_CHAT_DEPLOYMENT")  
+        self.azure_openai_key = os.getenv("AZURE_OPENAI_API_KEY")  
+        self.azure_openai_endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")  
+        self.api_version = os.getenv("AZURE_OPENAI_API_VERSION")  
+        self.mcp_server_uri = os.getenv("MCP_SERVER_URI") 
+        self.openai_model_name = os.getenv("OPENAI_MODEL_NAME")  
+  
+        self.session_id = session_id  
+        self.state_store = state_store  
+  
+        self.chat_history: List[Dict[str, str]] = self.state_store.get(f"{session_id}_chat_history", [])  
+        self.state: Optional[Any] = self.state_store.get(session_id, None) 
+        logging.debug(f"Chat history for session {session_id}: {self.chat_history}")  
+  
+    def _setstate(self, state: Any) -> None:  
+        self.state_store[self.session_id] = state  
+  
+    def append_to_chat_history(self, messages: List[Dict[str, str]]) -> None:  
+        self.chat_history.extend(messages)  
+        self.state_store[f"{self.session_id}_chat_history"] = self.chat_history  
+  
+    async def chat_async(self, prompt: str) -> str:  
+        """  
+        Override in child class!  
+        """  
+        raise NotImplementedError("chat_async should be implemented in subclass.")  

agentic_ai/agents/semantic_kernel/multi_agent/base_agent.py

@@ -344,17 +344,17 @@ async def chat_async(self, prompt: str) -> str:
 
 
 # --------------------------- Manual test helper --------------------------- #
-if __name__ == "__main__":
-
-    async def _demo() -> None:
-        dummy_state: dict = {}
-        agent = Agent(dummy_state, session_id="demo")
-        user_question = "My customer id is 101, what is my current balance?"
-        answer = await agent.chat_async(user_question)
-        print("\n>>> Assistant reply:\n", answer)
-        try:
-            await agent.contoso_plugin.close()
-        except Exception as exc:
-            logger.warning(f"SSE plugin close failed: {exc}")
-
-    asyncio.run(_demo())
+# if __name__ == "__main__":
+
+#     async def _demo() -> None:
+#         dummy_state: dict = {}
+#         agent = Agent(dummy_state, session_id="demo")
+#         user_question = "My customer id is 101, why is my internet bill so high?"
+#         answer = await agent.chat_async(user_question)
+#         print("\n>>> Assistant reply:\n", answer)
+#         try:
+#             await agent.contoso_plugin.close()
+#         except Exception as exc:
+#             logger.warning(f"SSE plugin close failed: {exc}")
+
+#     asyncio.run(_demo())