Class WorkersAiClient

java.lang.Object

dev.langchain4j.model.workersai.client.WorkersAiClient

public class WorkersAiClient
extends Object

Low level client to interact with the WorkerAI API.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	WorkersAiClient.Builder	Builder for WorkersAiClient.

Method Summary

Modifier and Type	Method	Description
static WorkersAiClient.Builder	builder()	Builder access.
WorkersAiEmbeddingResponse	embed(WorkersAiEmbeddingRequest apiRequest, String accountIdentifier, String modelName)	Generate embeddings.
WorkersAiChatCompletionResponse	generateChat(WorkersAiChatCompletionRequest apiRequest, String accountIdentifier, String modelName)	Generate chat.
byte[]	generateImage(WorkersAiImageGenerationRequest apiRequest, String accountIdentifier, String modelName)	Generate image.
WorkersAiTextCompletionResponse	generateText(WorkersAiTextCompletionRequest apiRequest, String accountIdentifier, String modelName)	Generate text.

Methods inherited from class Object

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

Method Details

generateChat

public WorkersAiChatCompletionResponse generateChat(WorkersAiChatCompletionRequest apiRequest,
 String accountIdentifier,
 String modelName)

Generate chat.

Parameters:

apiRequest - request.

accountIdentifier - account identifier.

modelName - model name.

Returns:

response.

generateText

public WorkersAiTextCompletionResponse generateText(WorkersAiTextCompletionRequest apiRequest,
 String accountIdentifier,
 String modelName)

Generate text.

Parameters:

apiRequest - request.

accountIdentifier - account identifier.

modelName - model name.

Returns:

response.

generateImage

public byte[] generateImage(WorkersAiImageGenerationRequest apiRequest,
 String accountIdentifier,
 String modelName)

Generate image. The endpoint returns the raw binary image, so the response bytes are returned as-is.

Parameters:

apiRequest - request.

accountIdentifier - account identifier.

modelName - model name.

Returns:

the raw image bytes.

embed

public WorkersAiEmbeddingResponse embed(WorkersAiEmbeddingRequest apiRequest,
 String accountIdentifier,
 String modelName)

Generate embeddings.

Parameters:

apiRequest - request.

accountIdentifier - account identifier.

modelName - model name.

Returns:

response.

builder

public static WorkersAiClient.Builder builder()

Builder access.

Returns:

builder instance