From 4f2825b5cbbd9da58024c1542a9eb1ed98c77f2b Mon Sep 17 00:00:00 2001 From: Sardor Taylakov Date: Thu, 20 Jul 2023 16:06:03 -0700 Subject: [PATCH] add-ability-to-configure-base-url useful to specify custom base URL for testing purposes with mock server in integration tests --- .../com/theokanning/openai/service/OpenAiService.java | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/service/src/main/java/com/theokanning/openai/service/OpenAiService.java b/service/src/main/java/com/theokanning/openai/service/OpenAiService.java index e1cb856f..1b5d3769 100644 --- a/service/src/main/java/com/theokanning/openai/service/OpenAiService.java +++ b/service/src/main/java/com/theokanning/openai/service/OpenAiService.java @@ -409,6 +409,15 @@ public static Retrofit defaultRetrofit(OkHttpClient client, ObjectMapper mapper) .build(); } + public static Retrofit customRetrofit(String baseUrl, OkHttpClient client, ObjectMapper mapper) { + return new Retrofit.Builder() + .baseUrl(baseUrl) + .client(client) + .addConverterFactory(JacksonConverterFactory.create(mapper)) + .addCallAdapterFactory(RxJava2CallAdapterFactory.create()) + .build(); + } + public Flowable mapStreamToAccumulator(Flowable flowable) { ChatFunctionCall functionCall = new ChatFunctionCall(null, null); ChatMessage accumulatedMessage = new ChatMessage(ChatMessageRole.ASSISTANT.value(), null);