A basic server and tools to test it.

.idea/drift-detector.iml

@@ -1,8 +1,10 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <module type="PYTHON_MODULE" version="4">
   <component name="NewModuleRootManager">
-    <content url="file://$MODULE_DIR$" />
-    <orderEntry type="jdk" jdkName="Python 3.10 (MCP-exp)" jdkType="Python SDK" />
+    <content url="file://$MODULE_DIR$">
+      <excludeFolder url="file://$MODULE_DIR$/.venv" />
+    </content>
+    <orderEntry type="jdk" jdkName="Python 3.10 (drift-detector)" jdkType="Python SDK" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
 </module>

.idea/misc.xml

@@ -1,4 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
-  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.10 (MCP-exp)" project-jdk-type="Python SDK" />
+  <component name="Black">
+    <option name="sdkName" value="Python 3.10 (MCP-exp)" />
+  </component>
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.10 (drift-detector)" project-jdk-type="Python SDK" />
 </project>

agent.py

@@ -0,0 +1,22 @@
+# agent.py
+import asyncio
+from mcp_agent.core.fastagent import FastAgent
+from mcp_agent.core.prompt import Prompt
+
+fast = FastAgent("Drift Test Agent")
+
+@fast.agent(
+    name="diagnostics",
+    instruction="Answer diagnostic questions to test LLM stability.",
+    servers=["drift-server"]
+)
+async def main():
+    async with fast.run() as agent:
+        # Apply prompt from the MCP server
+        print(">> Getting prompt from MCP server…")
+        result = await agent.apply_prompt("drift-diagnostics")
+        print(">> Response:")
+        print(result)
+
+if __name__ == "__main__":
+    asyncio.run(main())

fastagent.config.yaml

@@ -0,0 +1,11 @@
+mcp:
+  servers:
+    drift-server:
+      transport: stdio
+      command: "python"
+      args: ["server.py"]
+
+default_model: "generic.llama3.1"
+generic:
+  api_key: "ollama"  # doesn't matter, just a placeholder
+  base_url: "http://localhost:11434/v1"

server.py

@@ -0,0 +1,58 @@
+# main.py
+import asyncio
+from mcp.server import Server
+from mcp.server.stdio import stdio_server
+import mcp.types as types
+
+# Define diagnostic prompts statically for now
+PROMPTS = {
+    "drift-diagnostics": types.Prompt(
+        name="drift-diagnostics",
+        description="Run a diagnostic questionnaire to test LLM consistency.",
+        arguments=[],
+    )
+}
+
+# Setup server
+app = Server("mcp-drift-server", version="0.1.0")
+
+
+@app.list_prompts()
+async def list_prompts() -> list[types.Prompt]:
+    return list(PROMPTS.values())
+
+
+@app.get_prompt()
+async def get_prompt(name: str, arguments: dict[str, str] | None = None) -> types.GetPromptResult:
+    if name not in PROMPTS:
+        raise ValueError(f"Prompt not found: {name}")
+
+    # Static message for MVP – replace with dynamic question set later
+    return types.GetPromptResult(
+        messages=[
+            types.PromptMessage(
+                role="user",
+                content=types.TextContent(
+                    type="text",
+                    text="Answer the following: What's the capital of France?"
+                )
+            ),
+            types.PromptMessage(
+                role="user",
+                content=types.TextContent(
+                    type="text",
+                    text="Explain why the sky is blue."
+                )
+            ),
+        ]
+    )
+
+
+# Main entrypoint
+async def main():
+    async with stdio_server() as streams:
+        await app.run(streams[0], streams[1], app.create_initialization_options())
+
+
+if __name__ == "__main__":
+    asyncio.run(main())