From 0c76f2e6e7baeabb81c57d5de7baccfd4bc72a68 Mon Sep 17 00:00:00 2001
From: Adil Hafeez <adil.hafeez@gmail.com>
Date: Thu, 18 Sep 2025 11:08:51 -0700
Subject: [PATCH] more changs

---
 demos/use_cases/rag_agent/arch_config.yaml    |  7 ---
 .../rag_agent/response_generator_test.rest    | 40 ------------
 demos/use_cases/rag_agent/test.hurl           | 35 -----------
 demos/use_cases/rag_agent/test.rest           | 63 ++++++++++++-------
 4 files changed, 42 insertions(+), 103 deletions(-)
 delete mode 100644 demos/use_cases/rag_agent/src/rag_agent/response_generator_test.rest
 delete mode 100644 demos/use_cases/rag_agent/test.hurl

diff --git a/demos/use_cases/rag_agent/arch_config.yaml b/demos/use_cases/rag_agent/arch_config.yaml
index 1a50f36d..759ea621 100644
--- a/demos/use_cases/rag_agent/arch_config.yaml
+++ b/demos/use_cases/rag_agent/arch_config.yaml
@@ -13,7 +13,6 @@ agents:
   - name: research_agent
     kind: openai
     endpoint: http://host.docker.internal:10503
-
 listeners:
   - name: rag agent
     router: arch_agent_v2
@@ -25,12 +24,6 @@ listeners:
           - query_rewriter
           - context_builder
           - response_generator
-      - name: research_agent
-        description: deep research agent to handle complex queries that require fetching information from multiple sources
-        filter_chain:
-          - query_rewriter
-          - context_builder
-          - response_generator
     protocol: openai
     address: 0.0.0.0
     port: 8001
diff --git a/demos/use_cases/rag_agent/src/rag_agent/response_generator_test.rest b/demos/use_cases/rag_agent/src/rag_agent/response_generator_test.rest
deleted file mode 100644
index 3509f043..00000000
--- a/demos/use_cases/rag_agent/src/rag_agent/response_generator_test.rest
+++ /dev/null
@@ -1,40 +0,0 @@
-@baseUrl = http://0.0.0.0:10502
-@model = gpt-4o
-
-###
-
-# Health Check
-GET {{baseUrl}}/health
-
-###
-
-# Test 1: Simple Non-Streaming Chat Completion
-POST {{baseUrl}}/v1/chat/completions
-Content-Type: application/json
-
-{
-  "model": "{{model}}",
-  "messages": [
-    {
-      "role": "user",
-      "content": "Hello! Can you help me understand what machine learning is?"
-    }
-  ]
-}
-
-###
-
-# Test 2: Simple Streaming Chat Completion
-POST {{baseUrl}}/v1/chat/completions
-Content-Type: application/json
-
-{
-  "model": "{{model}}",
-  "messages": [
-    {
-      "role": "user",
-      "content": "Explain the concept of artificial intelligence in simple terms."
-    }
-  ],
-  "stream": true
-}
diff --git a/demos/use_cases/rag_agent/test.hurl b/demos/use_cases/rag_agent/test.hurl
deleted file mode 100644
index f8530ff6..00000000
--- a/demos/use_cases/rag_agent/test.hurl
+++ /dev/null
@@ -1,35 +0,0 @@
-# Step 1: Initialize session
-POST http://localhost:10101/mcp
-Content-Type: application/json
-Accept: application/json, text/event-stream
-
-{
-  "jsonrpc": "2.0",
-  "id": 1,
-  "method": "initialize",
-  "params": {
-    "protocolVersion": "2024-11-05",
-    "capabilities": {},
-    "clientInfo": {
-      "name": "ExampleClient",
-      "version": "1.0.0"
-    }
-  }
-}
-
-HTTP 200
-[Captures]
-session_id: header "mcp-session-id"
-
-# # Step 2: List tools (use session ID from previous response)
-POST http://localhost:10101/mcp
-Content-Type: application/json
-Accept: application/json, text/event-stream
-mcp-session-id: 07603206a9b44a3d91d76f6b16f24faa
-
-{
-  "jsonrpc": "2.0",
-  "id": 2,
-  "method": "tools/list",
-  "params": {}
-}
diff --git a/demos/use_cases/rag_agent/test.rest b/demos/use_cases/rag_agent/test.rest
index 64a33964..730ad9b2 100644
--- a/demos/use_cases/rag_agent/test.rest
+++ b/demos/use_cases/rag_agent/test.rest
@@ -1,31 +1,52 @@
-### Step 1: Initialize session
-POST http://localhost:10101/mcp
+@baseUrl = http://0.0.0.0:10502
+@model = gpt-4o
+
+# Health Check
+GET {{baseUrl}}/health
+
+###
+
+# Test 1: Simple Non-Streaming Chat Completion
+POST {{baseUrl}}/v1/chat/completions
 Content-Type: application/json
-Accept: application/json, text/event-stream
 
 {
-  "jsonrpc": "2.0",
-  "id": 1,
-  "method": "initialize",
-  "params": {
-    "protocolVersion": "2024-11-05",
-    "capabilities": {},
-    "clientInfo": {
-      "name": "ExampleClient",
-      "version": "1.0.0"
+  "model": "{{model}}",
+  "messages": [
+    {
+      "role": "user",
+      "content": "Hello! Can you help me understand what machine learning is?"
     }
-  }
+  ]
 }
 
-### Step 2: List tools (use session ID from previous response)
-POST http://localhost:10101/mcp
+###
+
+# Test 2: Simple Streaming Chat Completion
+POST {{baseUrl}}/v1/chat/completions
 Content-Type: application/json
-Accept: application/json, text/event-stream
-mcp-session-id: af2e2dace64c48f99ac3536faeaa3c68
 
 {
-  "jsonrpc": "2.0",
-  "id": 2,
-  "method": "tools/list",
-  "params": {}
+  "model": "{{model}}",
+  "messages": [
+    {
+      "role": "user",
+      "content": "Explain the concept of artificial intelligence in simple terms."
+    }
+  ],
+  "stream": true
+}
+
+### Test 3
+POST http://localhost:8001/v1/chat/completions
+Content-Type: application/json
+
+{
+  "model": "{{model}}",
+  "messages": [
+    {
+      "role": "user",
+      "content": "what is the effective date of the master agreement for arcadyan"
+    }
+  ]
 }