WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-13 05:07:51 +02:00
Code Issues Packages Projects Releases Wiki Activity

imp

Browse Source
This commit is contained in:
Quanzheng Long
2026-03-12 21:48:32 -07:00
parent 896c1a8054
commit 03ad7a011c
2 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
langgraph-go/advancedgraph/graph.go
+3 -3
View File
@@ -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 {
langgraph-go/tests/sub_agents_test.go
+5 -6
View File
@@ -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])
}
}