diff --git a/langgraph-go/advancedgraph/graph.go b/langgraph-go/advancedgraph/graph.go index 84ade22e8..c38cd18d4 100644 --- a/langgraph-go/advancedgraph/graph.go +++ b/langgraph-go/advancedgraph/graph.go @@ -73,16 +73,16 @@ type resultOrErr struct { err error } -func (h *Handler) APublishToChannel(channel string, value any) error { +func (h *Handler) PublishToChannel(channel string, value any) error { return h.engine.Publish(channel, value) } -func (h *Handler) AResult() (map[string]any, error) { +func (h *Handler) WaitForResult() (map[string]any, error) { res := <-h.done return res.state, res.err } -func (g *CompiledGraph) AStart(initialState map[string]any) (*Handler, error) { +func (g *CompiledGraph) Start(initialState map[string]any) (*Handler, error) { engine := NewRustEngine() for _, ch := range g.asyncChannels { if err := engine.AddAsyncChannel(ch); err != nil { diff --git a/langgraph-go/tests/sub_agents_test.go b/langgraph-go/tests/sub_agents_test.go index ed1a72362..badd06e22 100644 --- a/langgraph-go/tests/sub_agents_test.go +++ b/langgraph-go/tests/sub_agents_test.go @@ -20,7 +20,7 @@ type mockPlanner struct { i int } -func (m *mockPlanner) ainvoke() []decision { +func (m *mockPlanner) invoke() []decision { if m.i >= len(m.responses) { return []decision{} } @@ -50,7 +50,7 @@ func TestSubAgentsEquivalentFlow(t *testing.T) { graph.AddNode("llm_node", func(ctx *ag.Context, arg any) (ag.Command, error) { state := arg.(map[string]any) - decisions := planner.ainvoke() + decisions := planner.invoke() sends := make([]ag.Send, 0, 4) for _, d := range decisions { if d.Type == "end" { @@ -142,7 +142,7 @@ func TestSubAgentsEquivalentFlow(t *testing.T) { graph.SetEntryPoint("llm_node") graph.SetFinishPoint("order_food_node") - handler, err := graph.Compile().AStart( + handler, err := graph.Compile().Start( map[string]any{ "input": "help me get something for lunch", "output": []string{}, @@ -154,11 +154,11 @@ func TestSubAgentsEquivalentFlow(t *testing.T) { } time.Sleep(10 * time.Millisecond) - if err := handler.APublishToChannel("user_input_channel", "No spicy food please"); err != nil { + if err := handler.PublishToChannel("user_input_channel", "No spicy food please"); err != nil { t.Fatalf("publish failed: %v", err) } - result, err := handler.AResult() + result, err := handler.WaitForResult() if err != nil { t.Fatalf("result failed: %v", err) } @@ -192,4 +192,3 @@ func TestSubAgentsEquivalentFlow(t *testing.T) { t.Fatalf("unexpected last output: %#v", output[len(output)-1]) } } -