Skip to content

Commit 4f2a6ed

Browse files
authored
feat(jmanus): set JManus UA (alibaba#1979)
feat(jmanus): Set JManus UA
1 parent dd1b9b2 commit 4f2a6ed

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/main/java/com/alibaba/cloud/ai/example/manus/llm/LlmService.java

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
import org.springframework.web.client.RestClient;
4646
import org.springframework.web.reactive.function.client.WebClient;
4747

48+
import java.util.HashMap;
4849
import java.util.List;
4950
import java.util.Map;
5051
import java.util.concurrent.ConcurrentHashMap;
@@ -348,9 +349,11 @@ public OpenAiChatModel openAiChatModel(DynamicModelEntity dynamicModelEntity, Op
348349
defaultOptions.setTopP(dynamicModelEntity.getTopP());
349350
}
350351
Map<String, String> headers = dynamicModelEntity.getHeaders();
351-
if (headers != null) {
352-
defaultOptions.setHttpHeaders(headers);
352+
if (headers == null) {
353+
headers = new HashMap<>();
353354
}
355+
headers.put("User-Agent", "JManus/3.0.2-SNAPSHOT");
356+
defaultOptions.setHttpHeaders(headers);
354357
var openAiApi = openAiApi(restClientBuilderProvider.getIfAvailable(RestClient::builder),
355358
webClientBuilderProvider.getIfAvailable(WebClient::builder), dynamicModelEntity);
356359
OpenAiChatOptions options = OpenAiChatOptions.fromOptions(defaultOptions);

0 commit comments

Comments
 (0)