Forge Integration

The AI SDK can be used standalone in any Go project. When building a Forge application, the AI extension (github.com/xraph/forge/extensions/ai) provides a thin DI wrapper that registers ai-sdk services into Forge's Vessel container.

How It Works01

The Forge AI extension:

1. Reads AI configuration from the Forge app config.

2. Creates an LLMManager, StateStore, and VectorStore.

3. Registers them in the Vessel container for dependency injection.

4. Creates an AgentFactory and AgentManager for managing agents.

Services Registered02

When config.Training.Enabled is true, the extension also registers:

• ai.ModelTrainer

• ai.DataManager

• ai.PipelineManager

Using the Extension03

import (
    "github.com/xraph/forge"
    "github.com/xraph/forge/extensions/ai"
)

aiExt := ai.NewExtension()

app := forge.New(
    forge.WithAppName("my-ai-app"),
    forge.WithExtensions(aiExt),
)

ctx := context.Background()
if err := app.Start(ctx); err != nil {
    log.Fatal(err)
}
defer app.Stop(ctx)

Resolving AI Services04

Use Vessel to resolve ai-sdk services from the container:

import (
    "github.com/xraph/vessel"
    sdk "github.com/xraph/ai-sdk"
)

// By type (recommended)
llmManager, err := vessel.InjectType[sdk.LLMManager](app.Container())

// By name
llmManager, err := vessel.Resolve[sdk.LLMManager](app.Container(), "forge.ai.sdk.llmManager")

Using in Extensions05

func (e *MyExtension) Register(app forge.App) error {
    return vessel.ProvideConstructor(app.Container(), func(
        llmManager sdk.LLMManager,
        logger forge.Logger,
    ) (*MyAIService, error) {
        return &MyAIService{
            llm:    llmManager,
            logger: logger,
        }, nil
    })
}

Important Notes06

• The AI extension falls back to in-memory stores when configured backends fail to connect.

• LLM providers are not auto-registered from config -- they must be registered manually.

• No routes are mounted automatically; register AgentController routes yourself.

• Extension Health() currently returns nil and does not aggregate component checks.

Standalone vs. Forge07

Further Reading08

• AI Extension docs -- full configuration, DI lifecycle, HTTP API, and troubleshooting

• Vessel docs -- dependency injection container