feat: upgrade to claude 2.1

Sefaria · Feb 22, 2024 · bb23d6a · bb23d6a
1 parent 5804f59
commit bb23d6a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/app/translation/poc.py b/app/translation/poc.py
@@ -23,7 +23,7 @@ def translate_text(text: str, context: str = None):
     if context:
         task_prompt = f"<context>{context}</context>{task_prompt}"
     task_message = HumanMessage(content=task_prompt)
-    llm = ChatAnthropic(model="claude-2", temperature=0)
+    llm = ChatAnthropic(model="claude-2.1", temperature=0)
     response_message = llm([identity_message, task_message])
     translation = get_by_xml_tag(response_message.content, 'translation')
     if translation is None: