Create accountability_agent.py

accountability_agent.py

@@ -0,0 +1,22 @@
+from langchain_community.callbacks import get_openai_callback
+from langchain_core.prompts import PromptTemplate
+from langchain_openai import ChatOpenAI
+
+
+class AccountabilityAgent:
+    def __init__(self, prompt_template_path) -> None:
+        self._llm = ChatOpenAI(model="gpt-4-1106-preview", temperature=0.3)
+
+        with open(prompt_template_path) as f:
+            self._prompt = PromptTemplate(input_variables=["query", "synthesis"], template=f.read().strip())
+
+        self._chain = self._prompt | self._llm
+
+    def run(self, query: str, cynthesis: str):
+        with get_openai_callback() as cb:
+            accountability = self._chain.invoke({"query": query, "synthesis": cynthesis}).content.strip()
+
+            tokens = cb.total_tokens
+            cost = cb.total_cost
+
+        return {"tokens": tokens, "cost": cost}, accountability