help

mit-submit · Aug 30, 2024 · cecf6d2 · cecf6d2
1 parent c2677d3
commit cecf6d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/a2rchi/utils/config_loader.py b/a2rchi/utils/config_loader.py
@@ -23,7 +23,7 @@ def load_config(self):
 
             # change the model class parameter from a string to an actual class
             MODEL_MAPPING = {
-                "AnthropicLLM": AnthropicLLM
+                "AnthropicLLM": AnthropicLLM,
                 "OpenAILLM": OpenAILLM,
                 "DumbLLM": DumbLLM,
                 "LlamaLLM": LlamaLLM

diff --git a/environment.yml b/environment.yml
@@ -205,6 +205,7 @@ dependencies:
   - zstd=1.5.5=hc292b87_0
   - pip:
       - accelerate==0.21.0
+      - anthropic==0.7.3
       - backoff==2.2.1
       - bitsandbytes==0.41.1
       - chroma-hnswlib==0.7.2