From 4132d07bcbb84114ade3182350a151882cc76b03 Mon Sep 17 00:00:00 2001
From: Simon Weniger <simonweniger.sw@gmail.com>
Date: Tue, 19 Sep 2023 16:17:00 +0200
Subject: [PATCH] add new agent function

---
 Sources/Models/Agent.swift         |  38 +--
 Sources/Models/AgentDocument.swift |  22 --
 Sources/Models/AgentTool.swift     |  23 --
 Sources/Models/PredictAgent.swift  |  28 --
 Sources/Models/Request.swift       |  21 ++
 Sources/main.swift                 | 520 ++++++++---------------------
 6 files changed, 165 insertions(+), 487 deletions(-)
 delete mode 100644 Sources/Models/AgentDocument.swift
 delete mode 100644 Sources/Models/AgentTool.swift
 delete mode 100644 Sources/Models/PredictAgent.swift
 create mode 100644 Sources/Models/Request.swift

diff --git a/Sources/Models/Agent.swift b/Sources/Models/Agent.swift
index 951562a..febcc1e 100644
--- a/Sources/Models/Agent.swift
+++ b/Sources/Models/Agent.swift
@@ -10,29 +10,19 @@ import Foundation
 
 public struct Agent: Codable {
 
-    public var name: String
-    public var type: String
-    public var llm: LLM?
-    public var hasMemory: Bool?
-    public var promptId: String?
-
-    public init(name: String, type: String, llm: LLM? = nil, hasMemory: Bool? = nil, promptId: String? = nil) {
-        self.name = name
-        self.type = type
-        self.llm = llm
-        self.hasMemory = hasMemory
-        self.promptId = promptId
+	public var isActive: Bool
+	public var name: String
+	public var prompt: String?
+    public var llmModel: String
+    public var description: String
+    public var avatar: String?
+
+	public init(name: String, isActive: Bool, prompt: String? = nil, llmModel: String, description: String, avatar: String? = nil) {
+		self.isActive = isActive
+		self.name = name
+        self.prompt = prompt
+        self.llmModel = llmModel
+        self.description = description
+        self.avatar = avatar
     }
-
-	public struct LLM: Codable {
-			public var provider: String
-			public var model: String
-			public var apiKey: String
-
-			public init(provider: String, model: String, apiKey: String) {
-				self.provider = provider
-				self.model = model
-				self.apiKey = apiKey
-			}
-		}
 }
diff --git a/Sources/Models/AgentDocument.swift b/Sources/Models/AgentDocument.swift
deleted file mode 100644
index 73f6626..0000000
--- a/Sources/Models/AgentDocument.swift
+++ /dev/null
@@ -1,22 +0,0 @@
-//
-// AgentDocument.swift
-//
-//  Created by Simon Weniger on 09.07.23.
-//
-
-import Foundation
-
-
-
-public struct AgentDocument: Codable {
-
-    public var agentId: String
-    public var documentId: String
-
-    public init(agentId: String, documentId: String) {
-        self.agentId = agentId
-        self.documentId = documentId
-    }
-
-
-}
diff --git a/Sources/Models/AgentTool.swift b/Sources/Models/AgentTool.swift
deleted file mode 100644
index 3ccb7db..0000000
--- a/Sources/Models/AgentTool.swift
+++ /dev/null
@@ -1,23 +0,0 @@
-//
-// AgentTool.swift
-//
-//  Created by Simon Weniger on 09.07.23.
-
-//
-
-import Foundation
-
-
-
-public struct AgentTool: Codable {
-
-    public var agentId: String
-    public var toolId: String
-
-    public init(agentId: String, toolId: String) {
-        self.agentId = agentId
-        self.toolId = toolId
-    }
-
-
-}
diff --git a/Sources/Models/PredictAgent.swift b/Sources/Models/PredictAgent.swift
deleted file mode 100644
index 15a235d..0000000
--- a/Sources/Models/PredictAgent.swift
+++ /dev/null
@@ -1,28 +0,0 @@
-//
-// PredictAgent.swift
-//
-//  Created by Simon Weniger on 09.07.23.
-//
-
-import Foundation
-
-
-public struct PredictAgent {
-	
-	public var input: [String:String]
-    public var hasStreaming: Bool?
-	public var session: String?
-
-	public init(input: [String:String], hasStreaming: Bool? = nil, session: String) {
-        self.input = input
-        self.hasStreaming = hasStreaming
-		self.session = session
-    }
-
-    public enum CodingKeys: String, CodingKey { 
-        case input
-        case hasStreaming = "has_streaming"
-		case session
-    }
-
-}
diff --git a/Sources/Models/Request.swift b/Sources/Models/Request.swift
new file mode 100644
index 0000000..739f5d0
--- /dev/null
+++ b/Sources/Models/Request.swift
@@ -0,0 +1,21 @@
+//
+// PredictAgent.swift
+//
+//  Created by Simon Weniger (Aiden Technologies) on 09.07.23.
+//
+
+import Foundation
+
+public struct Request {
+	
+	public var input: String
+    public var sessionId: String?
+	public var enableStreaming: Bool
+
+	public init(input: String, sessionId: String? = "", enableStreaming: Bool) {
+        self.input = input
+        self.sessionId = sessionId
+		self.enableStreaming = enableStreaming
+    }
+
+}
diff --git a/Sources/main.swift b/Sources/main.swift
index fe36f99..4cb5873 100644
--- a/Sources/main.swift
+++ b/Sources/main.swift
@@ -81,182 +81,53 @@ public struct SuperagentSDK {
 		return response
 	}
 	
-	//Prompts
-	///Retuns a specific prompt
-	public func getPrompt(id: String) async throws -> [String: Any] {
-		let data = try await request(method: .get, endpoint: "/prompts/\(id)")
-		guard let responseData = data as? [String: Any],
-			  let promptData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToRetrieve
-		}
-		
-		Swift.print("getPrompt result: \(promptData)")
-		
-		//let decoder = JSONDecoder()
-		//let jsonDataEncoded = try JSONSerialization.data(withJSONObject: jsonData, options: [])
-		//let prompt = try decoder.decode(Prompt.self, from: jsonDataEncoded)
-		
-		return promptData
-	}
+	//MARK: - START VERSION 1.0
 	
-	///Delete prompt
-	public func deletePrompt(id: String) async throws -> [String: Any] {
-		let data = try await request(method: .delete, endpoint: "/prompts/\(id)")
-		guard let responseData = data as? [String: Any],
-			  let success = responseData["success"] as? [String: Any] else {
-			throw SuperagentError.failedToRetrieve
-		}
-		
-		Swift.print("deletePrompt result: \(success)")
-		
-		return success
-	}
 	
-	///Lists all prompts
-	public func listPrompts() async throws -> [[String: Any]] {
-		let data = try await request(method: .get, endpoint: "/prompts")
-		
-		guard let responseData = data as? [String: Any],
-			  let promptsData = responseData["data"] as? [[String: Any]] else {
-			throw SuperagentError.failedToRetrieve
-		}
-		
-		Swift.print("listPrompts result: \(promptsData)")
-		
-		//let decoder = JSONDecoder()
-		//var prompts = [Prompt]()
-		
-		//for json in jsonArray {
-		//	let jsonData = try JSONSerialization.data(withJSONObject: json, options: [])
-		//	let prompt = try decoder.decode(Prompt.self, from: jsonData)
-		//	prompts.append(prompt)
-		//}
-		
-		return promptsData
-	}
 	
-	///Patch a specific prompt
-	public func updatePrompt(promptId: String, newPrompt: Prompt) async throws -> [String: Any] {
-		let payload: [String: Any] = ["name": newPrompt.name, "input_variables": newPrompt.inputVariables, "template": newPrompt.template]
-		let data = try await request(method: .patch, endpoint: "/prompts/\(promptId)", data: payload)
-		
-		guard let responseData = data as? [String: Any],
-			  let promptData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToRetrieve
-		}
-		
-		Swift.print("updatePrompt result: \(promptData)")
-		
-		return promptData
-	}
-	
-	///Create a new prompt
-	public func createPrompt(prompt: Prompt) async throws -> [String: Any] {
-		let payload: [String: Any] = ["name": prompt.name, "input_variables": prompt.inputVariables, "template": prompt.template]
-		let data = try await request(method: .post, endpoint: "/prompts", data: payload)
-		
-		guard let responseData = data as? [String: Any],
-			  let promptData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToRetrieve
-		}
-		
-		Swift.print("createPrompt result: \(promptData)")
-		
-		return promptData
-	}
-	
-	//Documents
-	///Returns a specific document
-	public func getDocument(id: String) async throws -> [String: Any] {
-		let data = try await request(method: .get, endpoint: "/documents/\(id)")
-		
-		guard let responseData = data as? [String: Any],
-			  let documentData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToRetrieve
-		}
-		
-		Swift.print("getDocument result: \(documentData)")
-		
-		return documentData
-	}
+	//MARK: - Agent
 	
-	///Delete document
-	public func deleteDocument(id: String) async throws -> [String: Any] {
-		let data = try await request(method: .delete, endpoint: "/documents/\(id)")
-		
-		guard let responseData = data as? [String: Any],
-			  let documentData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToRetrieve
-		}
-		
-		Swift.print("deleteDocument result: \(documentData)")
-		
-		return documentData
-	}
-	
-	///Lists all documents
-	public func listDocuments() async throws -> [String: Any] {
-		let data = try await request(method: .get, endpoint: "/documents")
-		
-		guard let responseData = data as? [String: Any],
-			  let documentData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToRetrieve
-		}
-		
-		Swift.print("listDocuments result: \(documentData)")
-		
-		return documentData
-	}
-	
-	///Create a new document
-	public func createDocument(document: Document) async throws -> [String: Any] {
-		let payload: [String: Any] = ["name": document.name,
-									  "description" : document.description as Any,
-									  "content" : document.content,
-									  "splitter": ["type": document.splitter?.type as Any, "chunk_size": document.splitter?.chunkSize as Any, "chunk_overlap": document.splitter?.chunkOverlap as Any],
-									  "url": document.url as Any,
-									  "type": document.type,
-									  "authorization": document.authorization as Any]
-		let data = try await request(method: .post, endpoint: "/documents", data: payload)
+	//List
+	///List all agents
+	public func listAgents() async throws -> [String: Any] {
+		let data = try await request(method: .get, endpoint: "/agents")
 		
 		guard let responseData = data as? [String: Any],
-			  let documentData = responseData["data"] as? [String: Any] else {
+			  let agentData = responseData["data"] as? [String: Any] else {
 			throw SuperagentError.failedToRetrieve
 		}
 		
 #if DEBUG
-		Swift.print("createDocument result: \(documentData)")
+		Swift.print("listAgents result: \(agentData)")
 #endif
 		
-		return documentData
+		return agentData
 	}
 	
-	///Patch a document
-	public func updateDocument(id: String, document: Document) async throws -> [String: Any] {
-		let payload: [String: Any] = ["name": document.name,
-									  "description" : document.description as Any,
-									  "content" : document.content,
-									  "splitter": ["type": document.splitter?.type as Any, "chunk_size": document.splitter?.chunkSize as Any, "chunk_overlap": document.splitter?.chunkOverlap as Any],
-									  "url": document.url as Any,
-									  "type": document.type,
-									  "authorization": document.authorization as Any]
-		let data = try await request(method: .patch, endpoint: "/documents/\(id)", data: payload)
+	//Create
+	///Create a new agent
+	public func createAgent(agent: Agent) async throws -> [String: Any] {
+		let payload: [String: Any] = ["isActive": agent.isActive,
+									  "name": agent.name,
+									  "prompt": agent.prompt ?? "",
+									  "llmModel": agent.llmModel,
+									  "description": agent.description,
+									  "avatar": agent.avatar ?? ""]
+		let data = try await request(method: .post, endpoint: "/agents", data: payload)
 		
 		guard let responseData = data as? [String: Any],
-			  let documentData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToRetrieve
-			
+			  let agentData = responseData["data"] as? [String: Any] else {
+			throw SuperagentError.requestFailed
 		}
-		
 #if DEBUG
-		Swift.print("updateDocument result: \(documentData)")
+		Swift.print("createAgent result: \(agentData)")
 #endif
 		
-		return documentData
+		return agentData
 	}
 	
-	//Agents
-	///Returns a specific agent
+	//Get
+	///Get a single agent
 	public func getAgent(id: String) async throws -> [String: Any] {
 		let data = try await request(method: .get, endpoint: "/agents/\(id)")
 		
@@ -269,362 +140,231 @@ public struct SuperagentSDK {
 		Swift.print("getAgent result: \(agentData)")
 #endif
 		
-		//let decoder = JSONDecoder()
-		//let jsonDataEncoded = try JSONSerialization.data(withJSONObject: jsonData, options: [])
-		//let agent = try decoder.decode(Agent.self, from: jsonDataEncoded)
-		
 		return agentData
 	}
 	
-	///Delete agent
-	public func deleteAgent(id: String) async throws -> [String: Any] {
-		let data = try await request(method: .delete, endpoint: "/agents/\(id)")
+	//Update
+	///Patch an agent
+	public func updateAgent(agentId: String, newAgent: Agent) async throws -> [String: Any] {
+		let payload: [String: Any] = ["isActive": newAgent.isActive,
+									  "name": newAgent.name,
+									  "prompt": newAgent.prompt ?? "",
+									  "llmModel": newAgent.llmModel,
+									  "description": newAgent.description,
+									  "avatar": newAgent.avatar ?? ""]
+		let data = try await request(method: .post, endpoint: "/agents/\(agentId)", data: payload)
 		
 		guard let responseData = data as? [String: Any],
 			  let agentData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToRetrieve
+			throw SuperagentError.failedToUpdate
 		}
-		
 #if DEBUG
-		Swift.print("deleteAgent result: \(agentData)")
+		Swift.print("createAgent result: \(agentData)")
 #endif
 		
 		return agentData
 	}
 	
-	///Lists all agents
-	public func listAgents() async throws -> [String: Any] {
-		let data = try await request(method: .get, endpoint: "/agents")
+	///Delete agent
+	public func deleteAgent(id: String) async throws -> [String: Any] {
+		let data = try await request(method: .delete, endpoint: "/agents/\(id)")
 		
 		guard let responseData = data as? [String: Any],
 			  let agentData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToRetrieve
+			throw SuperagentError.requestFailed
 		}
 		
 #if DEBUG
-		Swift.print("listAgents result: \(agentData)")
+		Swift.print("deleteAgent result: \(agentData)")
 #endif
 		
 		return agentData
 	}
 	
-	///Create a new agent
-	public func createAgent(agent: Agent) async throws -> [String: Any] {
-		let payload: [String: Any] = ["name": agent.name,
-									  "llm": ["provider": agent.llm?.provider, "model": agent.llm?.model, "api_key": agent.llm?.apiKey],
-									  "type": agent.type,
-									  "hasMemory": agent.hasMemory as Any,
-									  "promptId": agent.promptId ?? ""]
-		let data = try await request(method: .post, endpoint: "/agents", data: payload)
+	//Invoke
+	///Invoke an agent
+	public func invokeAgent(agentId: String, agentRequest: Request) async throws -> String {
+		let payload: [String: Any] = ["input": agentRequest.input,
+									  "sessionId": agentRequest.sessionId as Any,
+									  "enableStreaming": agentRequest.enableStreaming as Any]
 		
-		guard let responseData = data as? [String: Any],
-			  let agentData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToRetrieve
-		}
-#if DEBUG
-		Swift.print("createAgent result: \(agentData)")
-#endif
-		// WE MAY ADD RESPONSE MODELS LATER AND ADD JSON DECODING
+		let data = try await request(method: .post, endpoint: "/agents/\(agentId)/invoke", data: payload)
 		
-		//print("response data:\(jsonData)")
-		
-		//let decoder = JSONDecoder()
-		//let jsonDataEncoded = try JSONSerialization.data(withJSONObject: jsonData, options: [])
-		
-		//print("JsonDataEncoded \(jsonDataEncoded)")
-		//let agent = try decoder.decode(Agent.self, from: jsonDataEncoded)
-		
-		//print("returned Agent: \(agent)")
-		
-		return agentData
-	}
-	
-	
-	///Create a new prediction
-	public func createPrediction(agentId: String, prediction: PredictAgent) async throws -> String {
-		//print("create Prediction called")
-		let payload: [String: Any] = ["input": prediction.input,
-									  "has_Streaming": prediction.hasStreaming as Any,
-									  "session": prediction.session as Any]
-
-#if DEBUG
-		Swift.print("Prediction payload: \(payload)")
-#endif
-		let data = try await request(method: .post, endpoint: "/agents/\(agentId)/predict", data: payload)
 #if DEBUG
 		Swift.print("Prediction data:\(data)")
 #endif
+		
 		guard let responseData = data as? [String: Any],
 			  let predictionData = responseData["data"] as? String else {
 			throw SuperagentError.failedToRetrieve
 		}
-		Swift.print("createPrediction result: \(predictionData)")
-		return predictionData
-	}
-	
-	//Agent Library
-	///Get all Agents from the Library
-	public func listLibraryAgents() async throws -> [[String: Any]] {
-		let data = try await request(method: .get, endpoint: "/agents/library")
 		
-		guard let responseData = data as? [String: Any],
-			  let agentLibrary = responseData["data"] as? [[String: Any]] else {
-			throw SuperagentError.failedToRetrieve
-		}
 #if DEBUG
-		Swift.print("getAgentDocuments result: \(agentLibrary)")
+		Swift.print("createPrediction result: \(predictionData)")
 #endif
-		return agentLibrary
-	}
-	
-	
-	//Agent Documents
-	///Get all Documents from an Agent
-	public func listAgentDocuments() async throws -> [[String: Any]] {
-		let data = try await request(method: .get, endpoint: "/agent-documents")
 		
-		guard let responseData = data as? [String: Any],
-			  let agentDocuments = responseData["data"] as? [[String: Any]] else {
-			throw SuperagentError.failedToRetrieve
-		}
-#if DEBUG
-		Swift.print("getAgentDocuments result: \(agentDocuments)")
-#endif
-		return agentDocuments
+		return predictionData
 	}
 	
-	///Get all Documents from an Agent
-	public func getAgentDocument(agentDocumentId: String) async throws -> [String: Any] {
-		let data = try await request(method: .get, endpoint: "/agent-documents\(agentDocumentId)")
+	//Add LLM
+	///Add LLM to agent
+	public func addLlmToAgent(agentId: String, llmId: String) async throws -> String {
+		let data = try await request(method: .post, endpoint: "/agents/\(agentId)/llms", data: ["llmId": llmId])
 		
 		guard let responseData = data as? [String: Any],
-			  let agentDocument = responseData["data"] as? [String: Any] else {
+			  let agentData = responseData["data"] as? String else {
 			throw SuperagentError.failedToRetrieve
 		}
-#if DEBUG
-		Swift.print("getAgentDocuments result: \(agentDocument)")
-#endif
-		return agentDocument
-	}
-	
-	///Add a Document to an Agent
-	public func addDocumentToAgent(documentId: String, agentId: String) async throws -> [String: Any] {
-		let payload: [String: Any] = ["documentId": documentId, "agentId": agentId]
-		let data = try await request(method: .post, endpoint: "/agents", data: payload)
 		
-		guard let responseData = data as? [String: Any],
-			  let agentDocument = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToCreate
-		}
 #if DEBUG
-		Swift.print("createAgentDocument result: \(agentDocument)")
+		Swift.print("add LLM To Agent result: \(agentData)")
 #endif
-		return agentDocument
-	}
-	
-	///Delete a Document from an Agent
-	public func deleteAgentDocument(agentDocumentId: String) async throws -> [String: Any] {
-		let data = try await request(method: .delete, endpoint: "/agent-documents/\(agentDocumentId)")
 		
-		guard let responseData = data as? [String: Any],
-			  let agentDocument = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToRetrieve
-		}
-#if DEBUG
-		Swift.print("deleteAgentDocument result: \(agentDocument)")
-#endif
-		return agentDocument
+		return agentData
 	}
 	
-	//Agent Tools
-	///Get all Tools from an Agent
-	public func listAgentTools(agentToolId: String) async throws -> [String: Any] {
-		let data = try await request(method: .get, endpoint: "/agent-tools/\(agentToolId)")
+	//Remove LLM
+	///Remove LLM from agent
+	public func removeLlmfromAgent(agentId: String, llmId: String) async throws -> String {
+		let data = try await request(method: .delete, endpoint: "/agents/\(agentId)/llms/\(llmId)")
 		
 		guard let responseData = data as? [String: Any],
-			  let agentTool = responseData["data"] as? [String: Any] else {
+			  let agentData = responseData["data"] as? String else {
 			throw SuperagentError.failedToRetrieve
 		}
+		
 #if DEBUG
-		Swift.print("getAgentTools result: \(agentTool)")
+		Swift.print("remove LLM from Agent result: \(agentData)")
 #endif
-		return agentTool
+		
+		return agentData
 	}
 	
-	///Get all Tools from an Agent
-	public func getAgentTool() async throws -> [[String: Any]] {
-		let data = try await request(method: .get, endpoint: "/agent-tools")
+	//List Tools
+	///List agent tools
+	public func listAgentTools(agentId: String) async throws -> [String: Any] {
+		let data = try await request(method: .get, endpoint: "/agents/\(agentId)/tools")
 		
 		guard let responseData = data as? [String: Any],
-			  let agentTools = responseData["data"] as? [[String: Any]] else {
+			  let agentTools = responseData["data"] as? [String: Any] else {
 			throw SuperagentError.failedToRetrieve
 		}
 #if DEBUG
-		Swift.print("getAgentTools result: \(agentTools)")
+		Swift.print("list agent tools result: \(agentTools)")
 #endif
 		return agentTools
 	}
 	
-	///Add a Tool to an Agent
+	
+	//Add Tool
+	///Add tool to agent
 	public func addToolToAgent(agentId: String, toolId: String) async throws -> [String: Any] {
-		let payload: [String: Any] = ["agentId": agentId, "toolId": toolId]
-		let data = try await request(method: .post, endpoint: "/agent-tools", data: payload)
+		let data = try await request(method: .post, endpoint: "/agents/\(agentId)/tools", data: ["toolId": toolId])
 		
 		guard let responseData = data as? [String: Any],
-			  let agentToolData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToUpdate
+			  let agentTools = responseData["data"] as? [String: Any] else {
+			throw SuperagentError.failedToRetrieve
 		}
 #if DEBUG
-		Swift.print("addToolToAgent result: \(agentToolData)")
+		Swift.print("add tool to agent result: \(agentTools)")
 #endif
-		return agentToolData
+		return agentTools
 	}
 	
-	///Delete a Tool from an Agent
-	public func deleteAgentTool(agentToolId: String) async throws -> [String: Any] {
-		let data = try await request(method: .delete, endpoint: "/agent-documents/\(agentToolId)")
+	//Remove Tool
+	///Remove tool from agent
+	public func removeToolFromAgent(agentId: String, toolId: String) async throws -> [String: Any] {
+		let data = try await request(method: .delete, endpoint: "/agents/\(agentId)/tools/\(toolId)")
 		
 		guard let responseData = data as? [String: Any],
 			  let success = responseData["data"] as? [String: Any] else {
 			throw SuperagentError.requestFailed
 		}
 #if DEBUG
-		Swift.print("deleteAgentTool result: \(success)")
+		Swift.print("remove tool from agent result: \(success)")
 #endif
 		return success
 	}
 	
-	//Tools
-	///Returns a specific tool
-	public func getTool(id: String) async throws -> [String: Any] {
-		let data = try await request(method: .get, endpoint: "/tools/\(id)")
-		
-		guard let responseData = data as? [String: Any],
-			  let toolData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToRetrieve
-		}
-#if DEBUG
-		Swift.print("getTool result: \(toolData)")
-#endif
-		return toolData
-	}
-	
-	///Delete tool
-	public func deleteTool(id: String) async throws -> [String: Any] {
-		let data = try await request(method: .delete, endpoint: "/tools/\(id)")
+	//List Datasources
+	///List agent datasources
+	public func listAgentDatasources(agentId: String) async throws -> [String: Any] {
+		let data = try await request(method: .get, endpoint: "/agents/\(agentId)/datasources")
 		
 		guard let responseData = data as? [String: Any],
-			  let toolData = responseData["data"] as? [String: Any] else {
+			  let agentDatasources = responseData["data"] as? [String: Any] else {
 			throw SuperagentError.requestFailed
 		}
 #if DEBUG
-		Swift.print("deleteTool result: \(toolData)")
+		Swift.print("deleteAgentTool result: \(agentDatasources)")
 #endif
-		return toolData
+		return agentDatasources
 	}
 	
-	///Lists all tools
-	public func listTools() async throws -> [[String: Any]] {
-		let data = try await request(method: .get, endpoint: "/tools")
+	//Add Datasource
+	///Add datasource to agent
+	public func addDatasourceToAgent(agentId: String, datasourceId: String) async throws -> [String: Any] {
+		let data = try await request(method: .post, endpoint: "/agents/\(agentId)/datasources", data: ["datasourceId": datasourceId])
 		
 		guard let responseData = data as? [String: Any],
-			  let toolData = responseData["data"] as? [[String: Any]] else {
+			  let agentDatasources = responseData["data"] as? [String: Any] else {
 			throw SuperagentError.failedToRetrieve
 		}
 #if DEBUG
-		Swift.print("listTools result: \(toolData)")
+		Swift.print("add datasource to agent result: \(agentDatasources)")
 #endif
-		return toolData
+		return agentDatasources
 	}
 	
-	///Create a new tool
-	public func createTool(tool: Tool) async throws -> [String: Any] {
-		var payload: [String: Any] = ["name": tool.name, "type": tool.type]
-		if let metadata = tool.metadata {
-			payload["metadata"] = metadata
-		}
-		let data = try await request(method: .post, endpoint: "/tools", data: payload)
+	//Remove Datasource
+	///Remove datasource from agent
+	public func removeDatasourceFromAgent(agentId: String, datasourceId: String) async throws -> [String: Any] {
+		let data = try await request(method: .delete, endpoint: "/agents/\(agentId)/datasources/\(datasourceId)")
 		
 		guard let responseData = data as? [String: Any],
-			  let toolData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToCreate
+			  let success = responseData["data"] as? [String: Any] else {
+			throw SuperagentError.requestFailed
 		}
 #if DEBUG
-		Swift.print("createTool result: \(toolData)")
+		Swift.print("remove datasource from agent result: \(success)")
 #endif
-		return toolData
+		return success
 	}
 	
-	//Tags
-	///Returns a specific tag
-	public func getTag(id: String) async throws -> [String: Any] {
-		let data = try await request(method: .get, endpoint: "/tags/\(id)")
-		
-		guard let responseData = data as? [String: Any],
-			  let tagData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToRetrieve
-		}
-#if DEBUG
-		Swift.print("getTag result: \(tagData)")
-#endif
-		return tagData
-	}
-
-	///Delete tool
-	public func deleteTag(id: String) async throws -> [String: Any] {
-		let data = try await request(method: .delete, endpoint: "/tags/\(id)")
+	//List Runs
+	///List agent runs
+	public func listRuns(agentId: String) async throws -> [String: Any] {
+		let data = try await request(method: .get, endpoint: "/agents/\(agentId)/runs")
 		
 		guard let responseData = data as? [String: Any],
-			  let tagData = responseData["data"] as? [String: Any] else {
+			  let agentRuns = responseData["data"] as? [String: Any] else {
 			throw SuperagentError.requestFailed
 		}
 #if DEBUG
-		Swift.print("deleteTag result: \(tagData)")
+		Swift.print("List agent runs result: \(agentRuns)")
 #endif
-		return tagData
+		return agentRuns
 	}
-
-	///Lists all tools
-	public func listTags() async throws -> [[String: Any]] {
-		let data = try await request(method: .get, endpoint: "/tags")
+	
+	//MARK: - LLM
+	
+	//List
+	///List all LLMs
+	public func listLLMs() async throws -> [String: Any] {
+		let data = try await request(method: .get, endpoint: "/llms")
 		
 		guard let responseData = data as? [String: Any],
-			  let tagsData = responseData["data"] as? [[String: Any]] else {
+			  let llms = responseData["data"] as? [String: Any] else {
 			throw SuperagentError.failedToRetrieve
 		}
-#if DEBUG
-		Swift.print("listTagss result: \(tagsData)")
-#endif
-		return tagsData
-	}
-	///Create a new tool
-	public func createTag(tag: Tag) async throws -> [String: Any] {
-		var payload: [String: Any] = ["name": tag.name, "type": tag.color, "userId": tag.userId ?? ""]
-		let data = try await request(method: .post, endpoint: "/tags", data: payload)
 		
-		guard let responseData = data as? [String: Any],
-			  let tagData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToCreate
-		}
 #if DEBUG
-		Swift.print("createTag result: \(tagData)")
+		Swift.print("list all LLMs result: \(llms)")
 #endif
-		return tagData
-	}
-	
-	///Patch a tag
-	public func updateTag(id: String, newTag: Tag) async throws -> [String: Any] {
-		var payload: [String: Any] = ["name": newTag.name, "type": newTag.color]
-		let data = try await request(method: .post, endpoint: "/tags/\(id)", data: payload)
 		
-		guard let responseData = data as? [String: Any],
-			  let tagData = responseData["data"] as? [String: Any] else {
-			throw SuperagentError.failedToUpdate
-		}
-#if DEBUG
-		Swift.print("updateTag result: \(tagData)")
-#endif
-		return tagData
+		return llms
 	}
-}
-
+	
+	//Create
+	///Create a new LLM
+