jproman's picture
Refactor to use langchain
522e6c3
raw
history blame contribute delete
649 Bytes
import os
os.environ["LOCALLLM"] = "llama3.2"
from flow import createFlow
from logger import Logger
import config
class Agent:
def __init__(self, logger):
self.logger = logger
def invoke(self, question):
flow = createFlow(self.logger)
shared = { "question": question }
flow.run(shared)
return shared["answer"]
if __name__ == "__main__":
from assignment_utils import getQuestionByPos
logger = Logger(config.logLevel,config.logFile)
agent = Agent(logger)
question = getQuestionByPos(0)
print(question)
response = agent.invoke(question['question'])
print(response)