symposium 0.4.0 → 0.4.1

package/models/AnthropicModel.js

@@ -0,0 +1,25 @@
+import Model from "../Model.js";
+import Anthropic from '@anthropic-ai/sdk';
+
+export default class AnthropicModel extends Model {
+	anthropic;
+	supports_tools = false;
+
+	getAnthropic() {
+		if (!this.anthropic)
+			this.anthropic = new Anthropic({apiKey: process.env.ANTHROPIC_API_KEY});
+
+		return this.anthropic;
+	}
+
+	async generate(thread, payload = {}, functions = []) {
+		const completion_payload = {
+			model: this.name,
+			messages: thread.getMessagesJson(),
+			...payload,
+		};
+
+		const message = await this.getAnthropic().messages.create(completion_payload);
+		return message.content;
+	}
+}

package/models/Claude3Haiku.js

@@ -0,0 +1,7 @@
+import AnthropicModel from "./AnthropicModel.js";
+
+export default class Claude3Haiku extends AnthropicModel {
+	name = 'claude-3-haiku-20240307';
+	label = 'claude-3-haiku';
+	tokens = 200000;
+}

package/models/Claude3Opus.js

@@ -0,0 +1,7 @@
+import AnthropicModel from "./AnthropicModel.js";
+
+export default class Claude3Opus extends AnthropicModel {
+	name = 'claude-3-opus-20240229';
+	label = 'claude-3-opus';
+	tokens = 200000;
+}

package/models/Claude3Sonnet.js

@@ -0,0 +1,7 @@
+import AnthropicModel from "./AnthropicModel.js";
+
+export default class Claude3Sonnet extends AnthropicModel {
+	name = 'claude-3-sonnet-20240229';
+	label = 'claude-3-sonnet';
+	tokens = 200000;
+}

package/models/Gpt4.js

@@ -1,6 +1,6 @@
 import OpenAIModel from "./OpenAIModel.js";
 
-export default class Gpt4  extends OpenAIModel {
+export default class Gpt4 extends OpenAIModel {
 	name = 'gpt-4';
 	label = 'gpt-4';
 	tokens = 8192;

package/package.json

@@ -1,12 +1,13 @@
 {
   "type": "module",
   "name": "symposium",
-  "version": "0.4.0",
+  "version": "0.4.1",
   "description": "Agents",
   "main": "index.js",
   "author": "Domenico Giambra",
   "license": "ISC",
   "dependencies": {
+    "@anthropic-ai/sdk": "^0.19.1",
     "@travio/redis": "^2.0.0",
     "openai": "^4.12.1",
     "tiktoken": "^1.0.10"