Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Minor test fixtures fixes

Browse Source
This commit is contained in:
Christian Tzolov
2024-09-23 23:03:56 +02:00
parent 501774925c
commit c98caf26d2
4 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
models/spring-ai-bedrock/src/test/java/org/springframework/ai/bedrock/llama/BedrockLlamaCreateRequestTests.java
View File

@@ -51,16 +51,16 @@ public class BedrockLlamaCreateRequestTests {
var request = client.createRequest(new Prompt("Test message content"));
assertThat(request.prompt()).isNotEmpty();
assertThat(request.temperature()).isEqualTo(66.6f);
assertThat(request.topP()).isEqualTo(0.66f);
assertThat(request.temperature()).isEqualTo(66.6);
assertThat(request.topP()).isEqualTo(0.66);
assertThat(request.maxGenLen()).isEqualTo(666);
request = client.createRequest(new Prompt("Test message content",
BedrockLlamaChatOptions.builder().withTemperature(99.9).withMaxGenLen(999).withTopP(0.99).build()));
assertThat(request.prompt()).isNotEmpty();
assertThat(request.temperature()).isEqualTo(99.9f);
assertThat(request.topP()).isEqualTo(0.99f);
assertThat(request.temperature()).isEqualTo(99.9);
assertThat(request.topP()).isEqualTo(0.99);
assertThat(request.maxGenLen()).isEqualTo(999);
}

2
models/spring-ai-moonshot/src/test/java/org/springframework/ai/moonshot/MoonshotChatCompletionRequestTest.java
View File

@@ -38,7 +38,7 @@ public class MoonshotChatCompletionRequestTest {
assertThat(request.messages()).hasSize(1);
assertThat(request.topP()).isEqualTo(1);
assertThat(request.temperature()).isEqualTo(0.7f);
assertThat(request.temperature()).isEqualTo(0.7);
assertThat(request.maxTokens()).isNull();
assertThat(request.stream()).isFalse();
}

2
spring-ai-docs/src/main/antora/modules/ROOT/pages/api/chat/groq-chat.adoc
View File

@@ -13,7 +13,7 @@ NOTE: The Groq API is not fully compatible with the OpenAI API.
Be aware for the following https://console.groq.com/docs/openai[compatability constrains].
Additionally, currently Groq doesn't support multimodal messages.
Check the https://github.com/spring-projects/spring-ai/blob/main/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/chat/GroqWithOpenAiChatModelIT.java[GroqWithOpenAiChatModelIT.java] tests
Check the https://github.com/spring-projects/spring-ai/blob/main/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/chat/proxy/GroqWithOpenAiChatModelIT.java[GroqWithOpenAiChatModelIT.java] tests
for examples of using Groq with Spring AI.
== Prerequisites

2
spring-ai-docs/src/main/antora/modules/ROOT/pages/api/chat/ollama-chat.adoc
View File

@@ -189,7 +189,7 @@ For this, you need to configure the OpenAI base URL to your Ollama instance: `sp
image::spring-ai-ollama-over-openai.jpg[Ollama OpenAI API compatibility, 800, 600, align="center"]
Check the link:https://github.com/spring-projects/spring-ai/blob/main/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/chat/OllamaWithOpenAiChatModelIT.java[OllamaWithOpenAiChatModelIT.java] tests for examples of using Ollama over Spring AI OpenAI.
Check the link:https://github.com/spring-projects/spring-ai/blob/main/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/chat/proxy/OllamaWithOpenAiChatModelIT.java[OllamaWithOpenAiChatModelIT.java] tests for examples of using Ollama over Spring AI OpenAI.
== Sample Controller