feat(custom-agent): add basic support for custom agent execution #51

Implement a new service for executing custom agents using OkHttpClient and add a new Python server example for mocking market agent responses. Also, add a Kotlin test case to verify the execution service.

Author: phodal

example/custom_agent/requirements.txt

@@ -0,0 +1,4 @@
+fastapi==0.110.0
+uvicorn~=0.27.1
+pydantic~=2.6.3
+starlette~=0.36.3

example/custom_agent/server.py

@@ -0,0 +1,36 @@
+from typing import List
+
+import uvicorn
+from fastapi import FastAPI
+from fastapi.exceptions import RequestValidationError
+from pydantic import BaseModel
+from starlette import status
+from starlette.requests import Request
+from starlette.responses import JSONResponse
+
+app = FastAPI()
+
+
+class Message(BaseModel):
+    role: str
+    message: str
+
+
+class Messages(BaseModel):
+    messages: List[Message]
+
+@app.post("/api/agent/market")
+def mock_market(messages: Messages):
+    return "I am a mock market agent for intention: " + messages.messages[0].message
+
+
+@app.exception_handler(RequestValidationError)
+async def validation_exception_handler(request: Request, exc: RequestValidationError):
+    exc_str = f'{exc}'.replace('\n', ' ').replace('   ', ' ')
+    print(f"{request}: {exc_str}")
+    content = {'status_code': 10422, 'message': exc_str, 'data': None}
+    return JSONResponse(content=content, status_code=status.HTTP_422_UNPROCESSABLE_ENTITY)
+
+
+if __name__ == "__main__":
+    uvicorn.run(app, port=8765)

src/main/kotlin/cc/unitmesh/devti/counit/CustomAgentExecutor.kt

@@ -6,6 +6,7 @@ import cc.unitmesh.devti.counit.model.CustomAgentConfig
 import cc.unitmesh.devti.llms.custom.CustomRequest
 import cc.unitmesh.devti.llms.custom.Message
 import com.intellij.openapi.components.Service
+import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.project.Project
 import kotlinx.serialization.encodeToString
 import kotlinx.serialization.json.Json
@@ -17,6 +18,7 @@ import okhttp3.RequestBody.Companion.toRequestBody
 @Service(Service.Level.PROJECT)
 class CustomAgentExecutor(val project: Project) {
     private var client = OkHttpClient()
+    private val logger = logger<CustomAgentExecutor>()
 
     fun execute(input: String, agent: CustomAgentConfig): String? {
         val customRequest = CustomRequest(listOf(Message("user", input)))
@@ -32,7 +34,9 @@ class CustomAgentExecutor(val project: Project) {
                 builder.addHeader("Content-Type", "application/json")
             }
 
-            null -> TODO()
+            null -> {
+                logger.info("No auth type found for agent ${agent.name}")
+            }
         }
 
         client = client.newBuilder().build()

src/test/kotlin/cc/unitmesh/devti/counit/CustomAgentExecutorTest.kt

@@ -0,0 +1,17 @@
+package cc.unitmesh.devti.counit
+
+import cc.unitmesh.devti.counit.model.CustomAgentConfig
+import com.intellij.testFramework.LightPlatformTestCase
+
+
+class CustomAgentExecutorTest : LightPlatformTestCase() {
+    fun testExecute() {
+        val customAgentExecutor = CustomAgentExecutor(project)
+        val response = customAgentExecutor.execute(
+            "test",
+            CustomAgentConfig("test", "test", "http://127.0.0.1:8765/api/agent/market")
+        )
+
+//        assertEquals("test", response)
+    }
+}