diff --git a/test_scripts/go/client_test_2.go b/test_scripts/go/client_test_2.go index 99287a6..0392244 100644 --- a/test_scripts/go/client_test_2.go +++ b/test_scripts/go/client_test_2.go @@ -1,57 +1,55 @@ -// package main - -// import ( -// "context" -// "fmt" -// "log" -// "time" - -// "github.com/runagent-dev/runagent-go/pkg/client" -// ) - -// func main() { -// fmt.Println("=== Minimal Streaming Example ===") - -// // Create client for streaming entrypoint -// agentClient, err := client.NewWithAddress( -// "841debad-7433-46ae-a0ec-0540d0df7314", // agentID -// "minimal_stream", // streaming entrypoint tag -// true, // local -// "localhost", // host -// 8450, // port -// ) -// if err != nil { -// log.Fatalf("Failed to create client: %v", err) -// } -// defer agentClient.Close() - -// ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute) -// defer cancel() - -// // Run streaming -// fmt.Println("Starting stream...") -// stream, err := agentClient.RunStream(ctx, map[string]interface{}{ -// "role": "user", -// "message": "Analyze the benefits of remote work for software teams", -// }) -// if err != nil { -// log.Fatalf("Failed to start stream: %v", err) -// } -// defer stream.Close() - -// // Read from stream -// for { -// data, hasMore, err := stream.Next(ctx) -// if err != nil { -// log.Printf("Stream error: %v", err) -// break -// } - -// if !hasMore { -// fmt.Println("Stream completed") -// break -// } - -// fmt.Printf("Received: %v\n", data) -// } -// } +package main + +import ( + "context" + "fmt" + "log" + "time" + + "github.com/runagent-dev/runagent-go/pkg/client" +) + +func main() { + fmt.Println("=== Minimal Streaming Example ===") + + // Create client for streaming entrypoint + agentClient, err := client.New( + "", // agentID + "minimal_stream", // streaming entrypoint tag + true, // local + ) + if err != nil { + log.Fatalf("Failed to create client: %v", err) + } + defer agentClient.Close() + + ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute) + defer cancel() + + // Run streaming + fmt.Println("Starting stream...") + stream, err := agentClient.RunStream(ctx, map[string]interface{}{ + "role": "user", + "message": "Analyze the benefits of remote work for software teams", + }) + if err != nil { + log.Fatalf("Failed to start stream: %v", err) + } + defer stream.Close() + + // Read from stream + for { + data, hasMore, err := stream.Next(ctx) + if err != nil { + log.Printf("Stream error: %v", err) + break + } + + if !hasMore { + fmt.Println("Stream completed") + break + } + + fmt.Printf("Received: %v\n", data) + } +} diff --git a/test_scripts/go/client_test_ag2_stream.go b/test_scripts/go/client_test_ag2_stream.go index ad35388..09b159d 100644 --- a/test_scripts/go/client_test_ag2_stream.go +++ b/test_scripts/go/client_test_ag2_stream.go @@ -1,57 +1,55 @@ -// package main - -// import ( -// "context" -// "fmt" -// "log" -// "time" - -// "github.com/runagent-dev/runagent-go/pkg/client" -// ) - -// func main() { -// fmt.Println("=== AG2 Invoke Streaming Example ===") - -// // Create client for streaming entrypoint -// agentClient, err := client.NewWithAddress( -// "6d9a6fb8-a58c-49de-92ef-cd64d53da932", // agentID -// "ag2_stream", // entrypoint tag -// true, // local -// "localhost", // host -// 8451, // port -// ) -// if err != nil { -// log.Fatalf("Failed to create client: %v", err) -// } -// defer agentClient.Close() - -// ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute) -// defer cancel() - -// // Run streaming -// fmt.Println("Starting AG2 invoke stream...") -// stream, err := agentClient.RunStream(ctx, map[string]interface{}{ -// "message": "The capital of bd is dhaka", -// "max_turns": 4, -// }) -// if err != nil { -// log.Fatalf("Failed to start stream: %v", err) -// } -// defer stream.Close() - -// // Read from stream -// for { -// data, hasMore, err := stream.Next(ctx) -// if err != nil { -// log.Printf("Stream error: %v", err) -// break -// } - -// if !hasMore { -// fmt.Println("Stream completed") -// break -// } - -// fmt.Printf("Received: %v\n", data) -// } -// } +package main + +import ( + "context" + "fmt" + "log" + "time" + + "github.com/runagent-dev/runagent-go/pkg/client" +) + +func main() { + fmt.Println("=== AG2 Invoke Streaming Example ===") + + // Create client for streaming entrypoint + agentClient, err := client.New( + "", // agentID + "ag2_stream", // entrypoint tag + true, // local + ) + if err != nil { + log.Fatalf("Failed to create client: %v", err) + } + defer agentClient.Close() + + ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute) + defer cancel() + + // Run streaming + fmt.Println("Starting AG2 invoke stream...") + stream, err := agentClient.RunStream(ctx, map[string]interface{}{ + "message": "The capital of bd is dhaka", + "max_turns": 4, + }) + if err != nil { + log.Fatalf("Failed to start stream: %v", err) + } + defer stream.Close() + + // Read from stream + for { + data, hasMore, err := stream.Next(ctx) + if err != nil { + log.Printf("Stream error: %v", err) + break + } + + if !hasMore { + fmt.Println("Stream completed") + break + } + + fmt.Printf("Received: %v\n", data) + } +} diff --git a/test_scripts/go/client_test_autogen_stream.go b/test_scripts/go/client_test_autogen_stream.go new file mode 100644 index 0000000..d825895 --- /dev/null +++ b/test_scripts/go/client_test_autogen_stream.go @@ -0,0 +1,77 @@ +// AutoGen Streaming Example +package main + +import ( + "context" + "fmt" + "log" + "time" + + "github.com/runagent-dev/runagent-go/pkg/client" +) + +func main() { + fmt.Println("=== AutoGen Streaming Example ===") + + // Create client for streaming entrypoint + agentClient, err := client.New( + "", // agentID + "autogen_token_stream", // entrypoint tag (changed from autogen_invoke to autogen_stream) + true, // local + ) + if err != nil { + log.Fatalf("Failed to create client: %v", err) + } + defer agentClient.Close() + + ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute) + defer cancel() + + // Run streaming + fmt.Println("Starting AutoGen streaming conversation...") + stream, err := agentClient.RunStream(ctx, map[string]interface{}{ + "task": "What is autogen and how does it work?", + }) + if err != nil { + log.Fatalf("Failed to start stream: %v", err) + } + defer stream.Close() + + // Read from stream + for { + data, hasMore, err := stream.Next(ctx) + if err != nil { + log.Printf("Stream error: %v", err) + break + } + + if !hasMore { + fmt.Println("Stream completed") + break + } + + // Handle different types of streaming data + if dataMap, ok := data.(map[string]interface{}); ok { + switch dataMap["type"] { + case "status": + fmt.Printf("Status: %v\n", dataMap["message"]) + case "message": + fmt.Printf("Turn %v - %s: %v\n", + dataMap["turn"], + dataMap["sender"], + dataMap["content"]) + case "summary": + fmt.Printf("Summary - Total turns: %v\n", dataMap["total_turns"]) + fmt.Printf("Final result: %v\n", dataMap["final_result"]) + case "error": + fmt.Printf("Error: %v\n", dataMap["message"]) + case "complete": + fmt.Println("Conversation completed successfully") + default: + fmt.Printf("Received: %v\n", data) + } + } else { + fmt.Printf("Received: %v\n", data) + } + } +} diff --git a/test_scripts/go/client_test_llamaindex_stream.go b/test_scripts/go/client_test_llamaindex_stream.go new file mode 100644 index 0000000..7a55709 --- /dev/null +++ b/test_scripts/go/client_test_llamaindex_stream.go @@ -0,0 +1,50 @@ +package main + +import ( + "context" + "fmt" + "log" + "time" + + "github.com/runagent-dev/runagent-go/pkg/client" +) + +func main() { + agentClient, err := client.New( + "", // agentID + "math_stream", // entrypoint tag + true, // local + ) + if err != nil { + log.Fatalf("Failed to create client: %v", err) + } + defer agentClient.Close() + + ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute) + defer cancel() + + // Start streaming + stream, err := agentClient.RunStream(ctx, map[string]interface{}{ + "math_query": "What is 2 * 2?", + }) + if err != nil { + log.Fatalf("Failed to start stream: %v", err) + } + defer stream.Close() + + // Read stream + for { + data, hasMore, err := stream.Next(ctx) + if err != nil { + log.Printf("Stream error: %v", err) + break + } + + if !hasMore { + fmt.Println("Stream completed") + break + } + + fmt.Printf("Received: %v\n", data) + } +}