Less iterations for Llama agent (#466)

gnosis · Sep 12, 2024 · 607857d · 607857d
1 parent ed3f034
commit 607857d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/prediction_market_agent/agents/microchain_agent/deploy.py b/prediction_market_agent/agents/microchain_agent/deploy.py
@@ -163,6 +163,8 @@ class DeployableMicrochainModifiableSystemPromptAgent3(
 ):
     task_description = AgentIdentifier.MICROCHAIN_AGENT_OMEN_LEARNING_3
     model = SupportedModel.llama_31_instruct
+    # Force less iterations, because Replicate's API allows at max 4096 input tokens.
+    n_iterations = 20
     description = "Microchain agent with 'just born' system prompt, and ability to adjust its own system prompt, version 3. Uses Llama 3.1 model."