Add claude param and docs

.env.example

@@ -1,6 +1,7 @@
 OPENAI_API_KEY=
 
 # Optional
+ANTHROPIC_API_KEY=
 SERPAPI_API_KEY=
 WOLFRAM_ALPHA_APPID=
 DATABASE_PROVIDER=  # sqlite | supabase

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:

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"

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