Update agents.py

agents.py

@@ -168,11 +168,11 @@ def build_graph(provider: str = "rwkv"):
         )
     elif provider == "rwkv":
         # --- BEGIN RWKV SETUP ---
-        title = "rwkv7-g1-2.9b-20250519-ctx4096"
+        title = "rwkv7-g1-1.5b-20250429-ctx4096"
         pth = hf_hub_download(repo_id="BlinkDL/rwkv7-g1", filename=f"{title}.pth")
         model_path = pth.replace(".pth", "")
 
-        raw_llm = RWKV(model=model_path, strategy='cpu fp32')
+        raw_llm = RWKV(model=model_path, strategy='cpu fp16')
         pipeline = PIPELINE(raw_llm, "rwkv_vocab_v20230424")
 
         class RWKVWithTools:
@@ -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=1000)
+                out_str = self.pipeline.generate(prompt, token_count=300)
 
                 print(f'Response: {out_str}')