Update agents.py

agents.py

@@ -14,6 +14,10 @@ from langchain_community.vectorstores import SupabaseVectorStore
 from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_core.tools import tool
 
+os.environ["RWKV_V7_ON"] = '1' # ==> enable RWKV-7 mode
+os.environ['RWKV_JIT_ON'] = '1' # '1' for better speed
+os.environ["RWKV_CUDA_ON"] = '0' # '1' to compile CUDA kernel (10x faster), requires c++ compiler & cuda libraries
+
 from huggingface_hub import hf_hub_download
 from rwkv.model import RWKV
 from rwkv.utils import PIPELINE, PIPELINE_ARGS
@@ -203,9 +207,15 @@ def build_graph(provider: str = "rwkv"):
                 )
 
                 prompt = header + convo
+
+                print(f'Prompt: {prompt}')
                 
                 # delegate to RWKV invoke()
-                return self.pipeline.generate(prompt, token_count=200)
+                out_str = self.pipeline.generate(prompt, token_count=200)
+
+                print(f'Response: {out_str}')
+
+                return out_str
 
         llm = RWKVWithTools(raw_llm, system_prompt=system_prompt)
         # --- END RWKV SETUP ---