Final_Assignment_Agents_Course

Paused

budivoy commited on Jul 2

Commit

f190d97

verified ·

1 Parent(s): 864d4e0

Update agents.py

Update token count

Files changed (1) hide show

agents.py CHANGED Viewed

@@ -211,7 +211,7 @@ def build_graph(provider: str = "rwkv"):
                 print(f'Prompt: {prompt}')
                 # delegate to RWKV invoke()
-                out_str = self.pipeline.generate(prompt, token_count=200)
                 print(f'Response: {out_str}')

                 print(f'Prompt: {prompt}')
                 # delegate to RWKV invoke()
+                out_str = self.pipeline.generate(prompt, token_count=1000)
                 print(f'Response: {out_str}')