|
30 | 30 | import com.azure.ai.openai.OpenAIAsyncClient;
|
31 | 31 | import com.azure.ai.openai.OpenAIClient;
|
32 | 32 | import com.azure.ai.openai.OpenAIClientBuilder;
|
33 |
| -import com.azure.ai.openai.models.ChatChoice; |
34 |
| -import com.azure.ai.openai.models.ChatCompletions; |
35 |
| -import com.azure.ai.openai.models.ChatCompletionsFunctionToolCall; |
36 |
| -import com.azure.ai.openai.models.ChatCompletionsFunctionToolDefinition; |
37 |
| -import com.azure.ai.openai.models.ChatCompletionsJsonResponseFormat; |
38 |
| -import com.azure.ai.openai.models.ChatCompletionsOptions; |
39 |
| -import com.azure.ai.openai.models.ChatCompletionsResponseFormat; |
40 |
| -import com.azure.ai.openai.models.ChatCompletionsTextResponseFormat; |
41 |
| -import com.azure.ai.openai.models.ChatCompletionsToolCall; |
42 |
| -import com.azure.ai.openai.models.ChatCompletionsToolDefinition; |
43 |
| -import com.azure.ai.openai.models.ChatMessageContentItem; |
44 |
| -import com.azure.ai.openai.models.ChatMessageImageContentItem; |
45 |
| -import com.azure.ai.openai.models.ChatMessageImageUrl; |
46 |
| -import com.azure.ai.openai.models.ChatMessageTextContentItem; |
47 |
| -import com.azure.ai.openai.models.ChatRequestAssistantMessage; |
48 |
| -import com.azure.ai.openai.models.ChatRequestMessage; |
49 |
| -import com.azure.ai.openai.models.ChatRequestSystemMessage; |
50 |
| -import com.azure.ai.openai.models.ChatRequestToolMessage; |
51 |
| -import com.azure.ai.openai.models.ChatRequestUserMessage; |
52 |
| -import com.azure.ai.openai.models.CompletionsFinishReason; |
53 |
| -import com.azure.ai.openai.models.ContentFilterResultsForPrompt; |
54 |
| -import com.azure.ai.openai.models.FunctionCall; |
55 |
| -import com.azure.ai.openai.models.FunctionDefinition; |
| 33 | +import com.azure.ai.openai.models.*; |
56 | 34 | import com.azure.core.util.BinaryData;
|
57 | 35 | import io.micrometer.observation.Observation;
|
58 | 36 | import io.micrometer.observation.ObservationRegistry;
|
@@ -387,7 +365,8 @@ ChatCompletionsOptions toAzureChatCompletionsOptions(Prompt prompt) {
|
387 | 365 | private List<ChatCompletionsFunctionToolDefinition> getFunctionTools(Set<String> functionNames) {
|
388 | 366 | return this.resolveFunctionCallbacks(functionNames).stream().map(functionCallback -> {
|
389 | 367 |
|
390 |
| - FunctionDefinition functionDefinition = new FunctionDefinition(functionCallback.getName()); |
| 368 | + ChatCompletionsFunctionToolDefinitionFunction functionDefinition = new ChatCompletionsFunctionToolDefinitionFunction( |
| 369 | + functionCallback.getName()); |
391 | 370 | functionDefinition.setDescription(functionCallback.getDescription());
|
392 | 371 | BinaryData parameters = BinaryData
|
393 | 372 | .fromObject(ModelOptionsUtils.jsonToMap(functionCallback.getInputTypeSchema()));
|
|
0 commit comments