From 49fd393a63b804c0ce8328131560248e137b3152 Mon Sep 17 00:00:00 2001
From: Alex Velikanov <alex25368@gmail.com>
Date: Wed, 17 May 2023 13:42:36 -0700
Subject: [PATCH] Add claude param and docs

---
 .env.example            | 1 +
 README.md               | 2 ++
 src/utils/model_name.py | 1 +
 src/utils/parameters.py | 6 ++++--
 4 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/.env.example b/.env.example
index c1e4c4f..b553618 100644
--- a/.env.example
+++ b/.env.example
@@ -1,6 +1,7 @@
 OPENAI_API_KEY=
 
 # Optional
+ANTHROPIC_API_KEY=
 SERPAPI_API_KEY=
 WOLFRAM_ALPHA_APPID=
 DATABASE_PROVIDER=  # sqlite | supabase
diff --git a/README.md b/README.md
index ebef6b0..fea9e2b 100644
--- a/README.md
+++ b/README.md
@@ -47,6 +47,8 @@ To change the world, all you need to do is:
 ## Setting up the Discord Integration
 Read through the dedicated [Discord setup docs](DISCORD.md)
 
+## Using with Anthropic Claude
+Make sure you have an `ANTHROPIC_API_KEY` in your env, then you can use `poetry run world --claude` which will run the world using `claude-v1` for some calls and `claude-v1-instant` for others.
 ## Contributing
 
 We enthusiastically welcome contributions to GPTeam! To contribute, please follow these steps:
diff --git a/src/utils/model_name.py b/src/utils/model_name.py
index 6eb5240..877fc23 100644
--- a/src/utils/model_name.py
+++ b/src/utils/model_name.py
@@ -5,3 +5,4 @@ class ChatModelName(Enum):
     TURBO = "gpt-3.5-turbo"
     GPT4 = "gpt-4"
     CLAUDE = "claude-v1"
+    CLAUDE_INSTANT = "claude-instant-v1"
diff --git a/src/utils/parameters.py b/src/utils/parameters.py
index 2153710..df3546e 100644
--- a/src/utils/parameters.py
+++ b/src/utils/parameters.py
@@ -21,10 +21,12 @@
 ANNOUNCER_DISCORD_TOKEN = os.getenv("ANNOUNCER_DISCORD_TOKEN")
 
 DEFAULT_SMART_MODEL = (
-    ChatModelName.TURBO if "--turbo" in sys.argv else ChatModelName.GPT4
+    ChatModelName.TURBO if "--turbo" in sys.argv else ChatModelName.CLAUDE if "--claude" in sys.argv else ChatModelName.GPT4
 )
 
-DEFAULT_FAST_MODEL = ChatModelName.TURBO
+DEFAULT_FAST_MODEL = (
+     ChatModelName.CLAUDE_INSTANT if "--claude" in sys.argv else ChatModelName.TURBO
+)
 
 
 # Tools