Function testGemini

cmd/testai/main-testai.go:360–409

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

}

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

	opts := &uctypes.AIOptsType{
		APIType:      uctypes.APIType_GoogleGemini,
		APIToken:     apiKey,
		Model:        model,
		MaxTokens:    8192,
		Capabilities: []string{uctypes.AICapabilityTools, uctypes.AICapabilityImages, uctypes.AICapabilityPdfs},
	}

	// 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 Google Gemini 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,
		SystemPrompt: []string{"You are a helpful assistant. Be concise and clear in your responses."},
	}
	err := aiusechat.WaveAIPostMessageWrap(ctx, sseHandler, aiMessage, chatOpts)
	if err != nil {
		fmt.Printf("Google Gemini streaming error: %v\n", err)
	}
}

func testT1(ctx context.Context) {
	tool := aiusechat.GetAdderToolDefinition()