Справка Langgraph: запись ожидаемых действий в узлах ⇐ Python

[Anonymous]

Я пытаюсь следовать документации по langgraph и создать базовый график, как показано ниже. Основное предостережение моего графика заключается в том, что у меня нет LLM, а я просто делаю вызовы API, которые объединяются в ответ. Проблема, с которой я столкнулся, заключается в написании ожидаемых действий для выполнения вызовов API в узлах графа.
ПРИМЕЧАНИЕ. Этот же код работает, если я изменю его на неожиданные аналоги.< /p>
Код минимального воспроизведения:

Код: Выделить всё

class State(TypedDict):
messages: Annotated[list[str], add_messages]

class ReturnNodeValue:
def __init__(self, assistantId: str):
self.assistantId = assistantId

async def __call__(self, state: State, **kwargs) -> Any:
response = await self.custom_api_call(state["messages"][-1], self.assistantId)
return {"messages": [response]}

async def custom_api_call(self, message: str, assistantId: str) -> str:
return await callApi(message, assistantId)

class GraphBuilder:
def __init__(self):
self.graph_builder = StateGraph(State)
self.buildGraph()

def buildGraph(self, assistantId1, assistantId2):
self.graph_builder.add_edge(START, "node1")
self.graph_builder.add_edge("node1", "node2")
self.graph_builder.add_edge("node2", END)

self.graph_builder.add_node("node1", ReturnNodeValue(assistantId1))
self.graph_builder.add_node("node2", ReturnNodeValue(assistantId2))

async def executeGraph(self):
try:
initial_state = {"messages": "Hellow"}
app = self.graph_builder.compile()
result = await app.ainvoke(initial_state)
# Error on above line: "Dispatcher.dispatch_forever..() got an unexpected keyword argument 'context'"
return result
except Exception as e:
print(f"An error occurred: {e}")
raise

graph = GraphBuilder()
graph.buildGraph("abc123", "abc234")
result = graph.executeGraph()

Ошибка: Dispatcher.dispatch_forever..() получил неожиданный аргумент ключевого слова «контекст».
Ошибка возникает, когда я нажимаю app.ainvoke( ), и, как упоминалось в примечании, код работает отлично, если я ничего не жду.
Буду очень признателен за любую помощь, спасибо

Подробнее здесь: https://stackoverflow.com/questions/792 ... s-in-nodes