Package ai.optfor.springopenaiapi

Class OpenAIApi

java.lang.Object

ai.optfor.springopenaiapi.OpenAIApi

public class OpenAIApi
extends Object

Constructor Summary

Constructors

Constructor	Description
OpenAIApi(String openaiKey, org.springframework.web.client.RestTemplate restTemplate, PromptCache promptCache)

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)
EmbeddingResponse	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

Constructor Details

OpenAIApi

public OpenAIApi(String openaiKey,
 org.springframework.web.client.RestTemplate restTemplate,
 PromptCache promptCache)

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 EmbeddingResponse embedding(String model,
 String content)

embedding

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