OpenAI compatible model factory for the Embabel Agent Framework
OpenAI-compatible model factory for the Embabel Agent Framework. Supports OpenAI and any OpenAI-compatible API (Azure OpenAI, local models, etc.).
Maven:
<dependency>
<groupId>com.embabel.agent</groupId>
<artifactId>embabel-agent-openai</artifactId>
<version>0.3.3</version>
</dependency>Gradle (Kotlin DSL):
implementation("com.embabel.agent:embabel-agent-openai:0.3.3")import com.embabel.agent.openai.OpenAiCompatibleModelFactory
import com.embabel.common.ai.model.PricingModel
import io.micrometer.observation.ObservationRegistry
import java.time.LocalDate
// Create factory
val factory = OpenAiCompatibleModelFactory(
baseUrl = null, // Uses OpenAI default
apiKey = "your-openai-api-key",
completionsPath = null,
embeddingsPath = null,
observationRegistry = ObservationRegistry.create()
)
// Create LLM service
val llmService = factory.openAiCompatibleLlm(
model = "gpt-4",
pricingModel = PricingModel.usdPer1MTokens(30.0, 60.0),
provider = "OpenAI",
knowledgeCutoffDate = LocalDate.of(2023, 4, 1)
)
// Create embedding service
val embeddingService = factory.openAiCompatibleEmbeddingService(
model = "text-embedding-3-small",
provider = "OpenAI"
)Install with Tessl CLI
npx tessl i tessl/maven-com-embabel-agent--embabel-agent-openai