Fix state for document agent (#173)

- moves state to data file - improvements for turning off mode once complete - notes for discovered bugs in flow
microsoft · Oct 25, 2024 · e4e7e71 · e4e7e71
1 parent f748061
commit e4e7e71
Show file tree

Hide file tree

Showing 3 changed files with 219 additions and 101 deletions.
diff --git a/assistants/prospector-assistant/assistant/agents/document/config_defaults.py b/assistants/prospector-assistant/assistant/agents/document/config_defaults.py
@@ -53,9 +53,9 @@ class ArtifactModel(BaseModel):
 # It can also help with pacing the conversation,
 # For example, here we have set an exact time limit of 10 turns which the agent will try to fill.
 resource_constraint = ResourceConstraint(
-    quantity=10,
+    quantity=3,
     unit=ResourceConstraintUnit.TURNS,
-    mode=ResourceConstraintMode.EXACT,
+    mode=ResourceConstraintMode.MAXIMUM,
 )
 
 __all__ = [