Upload app.py

app.py

@@ -142,7 +142,7 @@ def call_claude_api(prompt: str) -> str:
         # conversation = client_claude.completions.create(
         #     prompt=f"{HUMAN_PROMPT} {prompt}{AI_PROMPT}",
         #     model="claude-3-5-sonnet-20241022",
-        #     max_tokens_to_sample=1024,
+        #     max_tokens=1024,
         #     temperature=0.7,
         # )
         # return conversation.completion.strip()
@@ -154,7 +154,7 @@ def call_claude_api(prompt: str) -> str:
                 {"role": "system", "content": "You are a helpful assistant."},
                 {"role": "user",   "content": prompt},
             ],
-            max_tokens_to_sample=1024,
+            max_tokens=1024,
             temperature=0.7,
         )
         return response["completion"].strip()