Package ai.optfor.springopenaiapi

Class OpenAIApi

java.lang.Object

ai.optfor.springopenaiapi.OpenAIApi

public class OpenAIApi
extends Object

Field Summary

Fields

Modifier and Type	Field	Description
final org.springframework.web.client.RestTemplate	restTemplate

Constructor Summary

Constructors

Constructor	Description
OpenAIApi(String openaiKey)

Method Summary

Modifier and Type	Method	Description
ChatCompletionResponse	chat(String model, String prompt, String role, Integer maxTokens, double temperature)
ChatCompletionResponse	chat(String model, List<ChatMessage> chats, int maxTokens, double temperature)
EmbeddingData	embedding(String model, String content)
EmbeddingResponse	embedding(String model, List<String> content)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

restTemplate

public final org.springframework.web.client.RestTemplate restTemplate

Constructor Details

OpenAIApi

public OpenAIApi(String openaiKey)

Method Details

chat

public ChatCompletionResponse chat(String model,
 String prompt,
 String role,
 Integer maxTokens,
 double temperature)

chat

public ChatCompletionResponse chat(String model,
 List<ChatMessage> chats,
 int maxTokens,
 double temperature)

embedding

public EmbeddingData embedding(String model,
 String content)

embedding

public EmbeddingResponse embedding(String model,
 List<String> content)