Spaces:

subashdvorak
/

trygithubactions

Runtime error

trygithubactions / src /genai /context_analysis_agent /utils /nodes.py

Refactored context analysis

7a3f093 about 1 year ago

714 Bytes

from langchain_core.messages import SystemMessage
from src.genai.utils.models_loader import llm_gpt
from .prompts import introduction_prompt
from .state import ConversationFormatter, State


class IntroductionNode:
    def __init__(self):
        self.llm = llm_gpt

    def run(self, state:State):
        template = introduction_prompt
        messages = [SystemMessage(content=template)] + state["messages"]
        result = self.llm.with_structured_output(ConversationFormatter).invoke(messages)
        print('Response:', result)
        return {
        "messages": [{"role": "assistant", "content": result.response}],
        "response": result.response,
        "completion": result.completion,
    }

	from langchain_core.messages import SystemMessage
	from src.genai.utils.models_loader import llm_gpt
	from .prompts import introduction_prompt
	from .state import ConversationFormatter, State


	class IntroductionNode:
	def __init__(self):
	self.llm = llm_gpt

	def run(self, state:State):
	template = introduction_prompt
	messages = [SystemMessage(content=template)] + state["messages"]
	result = self.llm.with_structured_output(ConversationFormatter).invoke(messages)
	print('Response:', result)
	return {
	"messages": [{"role": "assistant", "content": result.response}],
	"response": result.response,
	"completion": result.completion,
	}