Polish

langchain4j · May 1, 2024 · 37ad3d5 · 37ad3d5
1 parent 8dd12fd
commit 37ad3d5
Show file tree

Hide file tree

Showing 6 changed files with 4 additions and 27 deletions.
diff --git a/docs/docs/integrations/language-models/open-ai.md b/docs/docs/integrations/language-models/open-ai.md
@@ -71,9 +71,9 @@ OpenAiChatModel model = OpenAiChatModel.builder()
     .proxy(...)
     .logRequests(...)
     .logResponses(...)
+    .logStreamingResponses(...)
     .tokenizer(...)
     .customHeaders(...)
-    .isStreaming(...)
     .build();
 ```
 See the description of some of the parameters above [here](https://platform.openai.com/docs/api-reference/chat/create).

diff --git a/...j-ollama/src/test/java/dev/langchain4j/model/ollama/OllamaOpenAiStreamingChatModelIT.java b/...j-ollama/src/test/java/dev/langchain4j/model/ollama/OllamaOpenAiStreamingChatModelIT.java
@@ -25,7 +25,6 @@ class OllamaOpenAiStreamingChatModelIT extends AbstractOllamaLanguageModelInfras
             .temperature(0.0)
             .logRequests(true)
             .logResponses(true)
-            .isStreaming(true)
             .build();
 
     @Test

diff --git a/langchain4j-open-ai/src/main/java/dev/langchain4j/model/openai/OpenAiChatModel.java b/langchain4j-open-ai/src/main/java/dev/langchain4j/model/openai/OpenAiChatModel.java
@@ -74,8 +74,7 @@ public OpenAiChatModel(String baseUrl,
                            Boolean logRequests,
                            Boolean logResponses,
                            Tokenizer tokenizer,
-                           Map<String, String> customHeaders,
-                           boolean isStreaming) {
+                           Map<String, String> customHeaders) {
 
         baseUrl = getOrDefault(baseUrl, OPENAI_URL);
         if (OPENAI_DEMO_API_KEY.equals(apiKey)) {
@@ -178,11 +177,7 @@ public int estimateTokenCount(List<ChatMessage> messages) {
     }
 
     public static OpenAiChatModel withApiKey(String apiKey) {
-        return withApiKey(apiKey, false);
-    }
-
-    public static OpenAiChatModel withApiKey(String apiKey, boolean isStreaming) {
-        return builder().apiKey(apiKey).isStreaming(isStreaming).build();
+        return builder().apiKey(apiKey).build();
     }
 
     public static OpenAiChatModelBuilder builder() {

diff --git a/...hain4j-open-ai/src/test/java/dev/langchain4j/model/openai/OpenAiStreamingChatModelIT.java b/...hain4j-open-ai/src/test/java/dev/langchain4j/model/openai/OpenAiStreamingChatModelIT.java
@@ -42,7 +42,6 @@ class OpenAiStreamingChatModelIT {
             .temperature(0.0)
             .logRequests(true)
             .logResponses(true)
-            .isStreaming(true)
             .build();
 
     StreamingChatLanguageModel visionModel = OpenAiChatModel.builder()
@@ -53,7 +52,6 @@ class OpenAiStreamingChatModelIT {
             .temperature(0.0)
             .logRequests(true)
             .logResponses(true)
-            .isStreaming(true)
             .build();
 
     ToolSpecification calculator = ToolSpecification.builder()
@@ -311,7 +309,6 @@ void should_execute_multiple_tools_in_parallel_then_stream_answer() throws Excep
                 .temperature(0.0)
                 .logRequests(true)
                 .logResponses(true)
-                .isStreaming(true)
                 .build();
 
         UserMessage userMessage = userMessage("2+2=? 3+3=?");
@@ -423,7 +420,6 @@ void should_stream_valid_json() throws Exception {
                 .responseFormat("json_object")
                 .logRequests(true)
                 .logResponses(true)
-                .isStreaming(true)
                 .build();
 
         // when
@@ -576,7 +572,6 @@ void should_use_enum_as_model_name() {
                 .modelName(GPT_3_5_TURBO)
                 .logRequests(true)
                 .logResponses(true)
-                .isStreaming(true)
                 .build();
 
         String question = "What is the capital of Germany?";

diff --git a/langchain4j/src/test/java/dev/langchain4j/service/StreamingAiServicesIT.java b/langchain4j/src/test/java/dev/langchain4j/service/StreamingAiServicesIT.java
@@ -39,17 +39,8 @@ static Stream<StreamingChatLanguageModel> models() {
                         .organizationId(System.getenv("OPENAI_ORGANIZATION_ID"))
                         .logRequests(true)
                         .logResponses(true)
-                        .isStreaming(true)
                         .build(),
-                Azure
-
-
-
-
-
-
-
-          .builder()
+                AzureOpenAiStreamingChatModel.builder()
                         .endpoint(System.getenv("AZURE_OPENAI_ENDPOINT"))
                         .apiKey(System.getenv("AZURE_OPENAI_KEY"))
                         .logRequestsAndResponses(true)
@@ -254,7 +245,6 @@ void should_execute_multiple_tools_sequentially_then_answer() throws Exception {
                 .modelName(GPT_3_5_TURBO_0613) // this model can only call tools sequentially
                 .temperature(0.0)
                 .logRequests(true)
-                .isStreaming(true)
                 .logResponses(true)
                 .build();
 
@@ -354,7 +344,6 @@ void should_execute_multiple_tools_in_parallel_then_answer() throws Exception {
                 .temperature(0.0)
                 .logRequests(true)
                 .logResponses(true)
-                .isStreaming(true)
                 .build();
 
         ChatMemory chatMemory = MessageWindowChatMemory.withMaxMessages(10);

diff --git a/langchain4j/src/test/java/dev/langchain4j/service/StreamingAiServicesWithToolsIT.java b/langchain4j/src/test/java/dev/langchain4j/service/StreamingAiServicesWithToolsIT.java
@@ -39,7 +39,6 @@ static Stream<StreamingChatLanguageModel> models() {
                         .temperature(0.0)
                         .logRequests(true)
                         .logResponses(true)
-                        .isStreaming(true)
                         .build(),
                 MistralAiStreamingChatModel.builder()
                         .apiKey(System.getenv("MISTRAL_AI_API_KEY"))