Function testAnthropic

cmd/testai/main-testai.go:310–358

(ctx context.Context, model, message string, tools []uctypes.ToolDefinition)

}

func testAnthropic(ctx context.Context, model, message string, tools []uctypes.ToolDefinition) {
	apiKey := os.Getenv("ANTHROPIC_APIKEY")
	if apiKey == "" {
		fmt.Println("Error: ANTHROPIC_APIKEY environment variable not set")
		os.Exit(1)
	}

	opts := &uctypes.AIOptsType{
		APIType:       uctypes.APIType_AnthropicMessages,
		APIToken:      apiKey,
		Model:         model,
		MaxTokens:     4096,
		ThinkingLevel: uctypes.ThinkingLevelMedium,
	}

	// Generate a chat ID
	chatID := uuid.New().String()

	// Convert to AIMessage format for WaveAIPostMessageWrap
	aiMessage := &uctypes.AIMessage{
		MessageId: uuid.New().String(),
		Parts: []uctypes.AIMessagePart{
			{
				Type: uctypes.AIMessagePartTypeText,
				Text: message,
			},
		},
	}

	fmt.Printf("Testing Anthropic streaming with WaveAIPostMessageWrap, model: %s\n", model)
	fmt.Printf("Message: %s\n", message)
	fmt.Printf("Chat ID: %s\n", chatID)
	fmt.Println("---")

	testWriter := &TestResponseWriter{}
	sseHandler := sse.MakeSSEHandlerCh(testWriter, ctx)
	defer sseHandler.Close()

	chatOpts := uctypes.WaveChatOpts{
		ChatId:   chatID,
		ClientId: uuid.New().String(),
		Config:   *opts,
		Tools:    tools,
	}
	err := aiusechat.WaveAIPostMessageWrap(ctx, sseHandler, aiMessage, chatOpts)
	if err != nil {
		fmt.Printf("Anthropic streaming error: %v\n", err)
	}
}

func testGemini(ctx context.Context, model, message string, tools []uctypes.ToolDefinition) {
	apiKey := os.Getenv("GOOGLE_APIKEY")