feat: add mock apis (All-Hands-AI#564)

preritt · Apr 2, 2024 · c992a9f · c992a9f
1 parent 16fe830
commit c992a9f
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/opendevin/mock/listen.py b/opendevin/mock/listen.py
@@ -27,7 +27,20 @@ def read_root():
 
 @app.get("/litellm-models")
 def read_llm_models():
-    return ["model-mock"]
+    return [
+        "gpt-4",
+        "gpt-4-turbo-preview",
+        "gpt-4-0314",
+        "gpt-4-0613",
+    ]
+
+@app.get("/litellm-agents")
+def read_llm_agents():
+    return [
+        "MonologueAgent",
+        "CodeActAgent",
+        "PlannerAgent",
+    ]
 
 if __name__ == "__main__":
     uvicorn.run(app, host="127.0.0.1", port=3000)