Add tests for covering logic with response analysis for google_ai

[vladzabolotnyi]

Description

self-explanatory, anthropic or openai can be used as reference

[probelabs]

Welcome & Thank You!

Hi @vladzabolotnyi, thank you for creating this issue. Adding test coverage is crucial for maintaining code quality, and your contribution here is much appreciated.

Understanding the Task

Based on your description, the goal is to add tests for the response analysis logic for the google_ai vendor, using the existing tests for openai and anthropic as a blueprint. This involves testing the extraction of token counts, model details, and other metadata from Google AI's API responses.

Getting Started

The primary logic for handling Google AI responses can be found in the vendors/googleai/ directory. The existing tests for other vendors provide a clear and effective pattern to follow:

• OpenAI Tests: vendors/openai/openai_test.go
• Anthropic Tests: vendors/anthropic/anthropic_test.go

A new test file, vendors/googleai/googleai_test.go, should be created. It should contain test cases for both standard and streaming responses, asserting that the AnalyzeResponse and AnalyzeStreamingResponse functions correctly parse mock Google AI API outputs.

Next Steps

1. Create a new file: vendors/googleai/googleai_test.go.
2. Implement test cases similar to those in the reference files, covering different response scenarios (e.g., successful responses, responses with different token counts, etc.).
3. Use mock JSON responses that mimic the actual structure of the Google AI API.
4. Submit a pull request with the new tests.

This seems like a great task for someone new to the project. We'll label it as a good first issue.

Please let us know if you have any questions!

References:

• vendors/googleai/googleai.go – Google AI vendor implementation.
• switches/response_extractor.go – Generic response extraction logic.
• vendors/openai/openai_test.go – Reference tests for OpenAI.
• vendors/anthropic/anthropic_test.go – Reference tests for Anthropic.

---

Powered by Visor from Probelabs

💡 TIP: You can chat with Visor using /visor ask <your question>