Create agents.py

agents.py

@@ -0,0 +1,32 @@
+from transformers import pipeline
+from prompts import *
+
+generator = pipeline(
+    "text-generation",
+    model="HuggingFaceH4/zephyr-7b-beta",
+    max_new_tokens=300,
+    temperature=0.7
+)
+
+def call_llm(prompt):
+    return generator(prompt)[0]["generated_text"]
+
+def market_agent(problem, memory):
+    prompt = MARKET_PROMPT.format(problem=problem, memory=memory)
+    return call_llm(prompt)
+
+def finance_agent(problem, memory):
+    prompt = FINANCE_PROMPT.format(problem=problem, memory=memory)
+    return call_llm(prompt)
+
+def risk_agent(problem, memory):
+    prompt = RISK_PROMPT.format(problem=problem, memory=memory)
+    return call_llm(prompt)
+
+def ethics_agent(problem, memory):
+    prompt = ETHICS_PROMPT.format(problem=problem, memory=memory)
+    return call_llm(prompt)
+
+def synthesiser_agent(problem, memory):
+    prompt = SYNTHESIS_PROMPT.format(problem=problem, memory=memory)
+    return call_llm(prompt)