Package dev.braintrust.api

Class BraintrustApiClient.InMemoryImpl

java.lang.Object

dev.braintrust.api.BraintrustApiClient.InMemoryImpl

All Implemented Interfaces:

BraintrustApiClient

Enclosing interface:

BraintrustApiClient

public static class BraintrustApiClient.InMemoryImpl
extends Object
implements BraintrustApiClient

Implementation for test doubling

Nested Class Summary

Nested classes/interfaces inherited from interface dev.braintrust.api.BraintrustApiClient

BraintrustApiClient.CreateDatasetRequest, BraintrustApiClient.CreateExperimentRequest, BraintrustApiClient.CreateProjectRequest, BraintrustApiClient.Dataset, BraintrustApiClient.DatasetEvent, BraintrustApiClient.DatasetFetchRequest, BraintrustApiClient.DatasetFetchResponse, BraintrustApiClient.DatasetList, BraintrustApiClient.Experiment, BraintrustApiClient.ExperimentList, BraintrustApiClient.Function, BraintrustApiClient.FunctionInvokeRequest, BraintrustApiClient.FunctionListResponse, BraintrustApiClient.HttpImpl, BraintrustApiClient.InMemoryImpl, BraintrustApiClient.InsertEventsRequest, BraintrustApiClient.InsertEventsResponse, BraintrustApiClient.LoginRequest, BraintrustApiClient.LoginResponse, BraintrustApiClient.OrganizationAndProjectInfo, BraintrustApiClient.OrganizationInfo, BraintrustApiClient.Project, BraintrustApiClient.ProjectList, BraintrustApiClient.Prompt, BraintrustApiClient.PromptData, BraintrustApiClient.PromptListResponse

Constructor Summary

Constructors

Constructor	Description
InMemoryImpl(BraintrustApiClient.OrganizationAndProjectInfo... organizationAndProjectInfos)
InMemoryImpl(List<BraintrustApiClient.OrganizationAndProjectInfo> organizationAndProjectInfos, List<BraintrustApiClient.Prompt> prompts)

Method Summary

Modifier and Type	Method	Description
BraintrustApiClient.DatasetFetchResponse	fetchDatasetEvents(String datasetId, BraintrustApiClient.DatasetFetchRequest request)	Fetch dataset events with pagination
Optional<BraintrustApiClient.Dataset>	getDataset(String datasetId)	Get dataset metadata by ID
Optional<BraintrustApiClient.Function>	getFunction(String projectName, String slug, String version)	Get a function by project name and slug, with optional version.
BraintrustApiClient.Experiment	getOrCreateExperiment(BraintrustApiClient.CreateExperimentRequest request)	Creates an experiment.
BraintrustApiClient.Project	getOrCreateProject(String projectName)	Creates or gets a project by name.
BraintrustApiClient.OrganizationAndProjectInfo	getOrCreateProjectAndOrgInfo(BraintrustConfig config)	Get project and org info for the given config.
Optional<BraintrustApiClient.Project>	getProject(String projectId)	Gets a project by ID.
Optional<BraintrustApiClient.OrganizationAndProjectInfo>	getProjectAndOrgInfo()	Get project and org info for the default project ID
Optional<BraintrustApiClient.OrganizationAndProjectInfo>	getProjectAndOrgInfo(String projectId)	Get project and org info for the given project ID
Optional<BraintrustApiClient.Prompt>	getPrompt(String projectName, String slug, String version)	Get a prompt by slug and optional version
Object	invokeFunction(String functionId, BraintrustApiClient.FunctionInvokeRequest request)	Invoke a function (scorer, prompt, or tool) by its ID.
BraintrustApiClient.LoginResponse	login()	Attempt Braintrust login
List<BraintrustApiClient.Dataset>	queryDatasets(String projectName, String datasetName)	Query datasets by project name and dataset name

Methods inherited from class java.lang.Object

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

Constructor Details

InMemoryImpl

public InMemoryImpl(BraintrustApiClient.OrganizationAndProjectInfo... organizationAndProjectInfos)

InMemoryImpl

public InMemoryImpl(List<BraintrustApiClient.OrganizationAndProjectInfo> organizationAndProjectInfos,
 List<BraintrustApiClient.Prompt> prompts)

Method Details

login

public BraintrustApiClient.LoginResponse login()

Description copied from interface: BraintrustApiClient

Attempt Braintrust login

Specified by:

login in interface BraintrustApiClient

Returns:

LoginResponse containing organization info

getOrCreateProject

public BraintrustApiClient.Project getOrCreateProject(String projectName)

Description copied from interface: BraintrustApiClient

Creates or gets a project by name.

Specified by:

getOrCreateProject in interface BraintrustApiClient

getProject

public Optional<BraintrustApiClient.Project> getProject(String projectId)

Description copied from interface: BraintrustApiClient

Gets a project by ID.

Specified by:

getProject in interface BraintrustApiClient

getOrCreateExperiment

public BraintrustApiClient.Experiment getOrCreateExperiment(BraintrustApiClient.CreateExperimentRequest request)

Description copied from interface: BraintrustApiClient

Creates an experiment.

Specified by:

getOrCreateExperiment in interface BraintrustApiClient

getProjectAndOrgInfo

public Optional<BraintrustApiClient.OrganizationAndProjectInfo> getProjectAndOrgInfo()

Description copied from interface: BraintrustApiClient

Get project and org info for the default project ID

Specified by:

getProjectAndOrgInfo in interface BraintrustApiClient

getProjectAndOrgInfo

public Optional<BraintrustApiClient.OrganizationAndProjectInfo> getProjectAndOrgInfo(String projectId)

Description copied from interface: BraintrustApiClient

Get project and org info for the given project ID

Specified by:

getProjectAndOrgInfo in interface BraintrustApiClient

getOrCreateProjectAndOrgInfo

public BraintrustApiClient.OrganizationAndProjectInfo getOrCreateProjectAndOrgInfo(BraintrustConfig config)

Description copied from interface: BraintrustApiClient

Get project and org info for the given config. Creating them if necessary

Specified by:

getOrCreateProjectAndOrgInfo in interface BraintrustApiClient

getPrompt

public Optional<BraintrustApiClient.Prompt> getPrompt(@Nonnull
 String projectName,
 @Nonnull
 String slug,
 @Nullable
 String version)

Description copied from interface: BraintrustApiClient

Get a prompt by slug and optional version

Specified by:

getPrompt in interface BraintrustApiClient

fetchDatasetEvents

public BraintrustApiClient.DatasetFetchResponse fetchDatasetEvents(String datasetId,
 BraintrustApiClient.DatasetFetchRequest request)

Description copied from interface: BraintrustApiClient

Fetch dataset events with pagination

Specified by:

fetchDatasetEvents in interface BraintrustApiClient

getDataset

public Optional<BraintrustApiClient.Dataset> getDataset(String datasetId)

Description copied from interface: BraintrustApiClient

Get dataset metadata by ID

Specified by:

getDataset in interface BraintrustApiClient

queryDatasets

public List<BraintrustApiClient.Dataset> queryDatasets(String projectName,
 String datasetName)

Description copied from interface: BraintrustApiClient

Query datasets by project name and dataset name

Specified by:

queryDatasets in interface BraintrustApiClient

getFunction

public Optional<BraintrustApiClient.Function> getFunction(@Nonnull
 String projectName,
 @Nonnull
 String slug,
 @Nullable
 String version)

Description copied from interface: BraintrustApiClient

Get a function by project name and slug, with optional version.

Specified by:

getFunction in interface BraintrustApiClient

Parameters:

projectName - the name of the project containing the function

slug - the unique slug identifier for the function

version - optional version identifier (transaction id or version string)

Returns:

the function if found

invokeFunction

public Object invokeFunction(@Nonnull
 String functionId,
 @Nonnull
 BraintrustApiClient.FunctionInvokeRequest request)

Description copied from interface: BraintrustApiClient

Invoke a function (scorer, prompt, or tool) by its ID.

Specified by:

invokeFunction in interface BraintrustApiClient

Parameters:

functionId - the ID of the function to invoke

request - the invocation request containing input, expected output, etc.

Returns:

the result of the function invocation