Agent and Mission GraphQL CRUD operations

2024-02-20 17:06:45 +04:00
parent e6f08d85fc
commit 619f199abe
13 changed files with 1744 additions and 59 deletions
--- a/src/app/api/graphql/schema.ts
+++ b/src/app/api/graphql/schema.ts
@@ -0,0 +1,113 @@
+const typeDefs = `#graphql
+    enum AgentTool {
+        DUCK_DUCK_GO_SEARCH
+        PUBMED
+        PYTHON_REPL
+        SEMANTIC_SCHOLER
+        STACK_EXCHANGE
+        WIKIDATA
+        WIKIPEDIA
+        YAHOO_FINANCE
+        YUOUTUBE_SEARCH
+    }
+
+    type Agent {
+        id: ID!
+        role: String!
+        goal: String!
+        backstory: String
+        tools: [AgentTool!]!
+        allowDelegation: Boolean!
+        verbose: Boolean!
+        image: String
+        missions: [Mission!]
+    }
+
+    input AgentInput {
+        id: ID!
+    }
+
+    type DeleteOutput {
+        deleted: Boolean!
+    }
+
+    type Task {
+        name: String!
+        description: String!
+        agent: Agent!
+    }
+
+    input TaskInput {
+        name: String!
+        description: String!
+        agent: Int
+    }
+
+    type Mission {
+        id: ID!
+        name: String!
+        crew: [Agent!]
+        tasks: [Task]
+        verbose: Boolean
+        process: MissionProcess
+        result: String
+    }
+
+    enum MissionProcess {
+        SEQUENTIAL
+        HIERARCHICAL
+    }
+
+    type Query {
+        agents(filter: String): [Agent!]!
+        agent(id: Int!): Agent
+        missions(filter: String): [Mission!]!
+        mission(id: Int!): Mission
+    }
+
+    type Mutation {
+        createAgent(
+            role: String!
+            goal: String!
+            backstory: String
+            tools: [AgentTool!] = []
+            allowDelegation: Boolean = false
+            verbose: Boolean = false
+        ): Agent!
+
+        updateAgent(
+            id: Int!
+            role: String
+            goal: String
+            backstory: String
+            tools: [AgentTool!]
+            allowDelegation: Boolean
+            verbose: Boolean
+        ): Agent!
+
+        deleteAgent(id: Int!): DeleteOutput
+
+        createMission(
+            name: String!
+            crew: [Int!] = []
+            tasks: [TaskInput!] = []
+            verbose: Boolean = false
+            process: MissionProcess =  "SEQUENTIAL"
+        ): Mission!
+
+        updateMission(
+            id: Int!
+            name: String
+            crew: [Int!]
+            tasks: [TaskInput!]
+            verbose: Boolean
+            process: MissionProcess
+        ): Mission
+
+        deleteMission(id: Int!): DeleteOutput
+
+        runMission(id: Int!): Mission
+    }
+`;
+
+export default typeDefs;