Adapt tests

superagent-ai · Jun 25, 2023 · 51929be · 51929be
1 parent 9544982
commit 51929be
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 15 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "superagentai-js",
-  "version": "0.0.11",
+  "version": "0.0.12",
   "types": "src/superagent.d.ts",
   "description": "Superagent.sh Javascript SDK",
   "main": "src/superagent.js",

diff --git a/src/superagent.d.ts b/src/superagent.d.ts
@@ -23,10 +23,8 @@ declare module 'superagentSDK' {
     name: string;
     llm: string;
     type?: string;
-    has_memory: boolean;
-    documentId: string;
+    hasMemory: boolean;
     promptId: string;
-    toolId: string;
   }
 
   interface Prediction {

diff --git a/src/superagent.js b/src/superagent.js
@@ -1,6 +1,6 @@
 export default class SuperagentSDK {
   constructor(authToken) {
-    this.baseUrl = "https://api.superagent.sh/api/v1";
+    this.baseUrl = "http://127.0.0.1:8000/api/v1";
     this.authToken = authToken;
   }
 
@@ -84,19 +84,15 @@ export default class SuperagentSDK {
         name,
         llm,
         type = "REACT",
-        has_memory,
-        document_id = null,
-        prompt_id = null,
-        tool_id = null,
+        hasMemory,
+        promptId = null,
       }) =>
         await this._request("POST", "/agents", {
           name,
           type,
           llm,
-          has_memory,
-          documentId: document_id,
-          promptId: prompt_id,
-          toolId: tool_id,
+          hasMemory,
+          promptId,
         }),
       predict: async ({ id, input, has_streaming }) =>
         await this._request("POST", `/agents/${id}/predict`, {

diff --git a/test/sdk.test.js b/test/sdk.test.js
@@ -139,7 +139,7 @@ describe("Agents", () => {
     const { success, data } = await superagent.agents().create({
       name: "Test agent",
       type: "REACT",
-      has_memory: true,
+      hasMemory: true,
       llm: { provider: "openai-chat", model: "gpt-3.5-turbo" },
     });
 
@@ -166,7 +166,7 @@ describe("Agents", () => {
   it("should run a single agent", async () => {
     const { success, data } = await superagent.agents().predict({
       id: agent.id,
-      input: { human_input: "hi" },
+      input: { input: "hi" },
       has_streaming: false,
     });