gemini

Documentation

Index

• func AgentToolsToGenAISchema(tools []agent.ToolDefinition) ([]*genai.Tool, error)
• func ConvertJSONSchemaToGenAISchema(js *jsonschema.Schema) *genai.Schema
• func GenAISchema[T any]() *genai.Schema
• func ToolToGenAISchema(tool agent.ToolDefinition) (*genai.Schema, error)
• func TypeToGenAISchema(t reflect.Type) *genai.Schema
• type Client
  • func New(client *genai.Client, modelName string) *Client
  • func (c *Client) CreateMessages(ctx context.Context, req agent.MessagesRequest, ...) (agent.MessagesResponse, error)

Functions

func AgentToolsToGenAISchema

func AgentToolsToGenAISchema(tools []agent.ToolDefinition) ([]*genai.Tool, error)

AgentToolsToGenAISchema converts a map of tools to a list of function definitions using genai.Schema for the function parameters.

func ConvertJSONSchemaToGenAISchema

func ConvertJSONSchemaToGenAISchema(js *jsonschema.Schema) *genai.Schema

ConvertJSONSchemaToGenAISchema converts a jsonschema.Schema to a genai.Schema.

func GenAISchema

func GenAISchema[T any]() *genai.Schema

GenAISchema converts a Go type to a genai.Schema representation. This is the equivalent of JSONSchema but for genai.Schema.

func ToolToGenAISchema

func ToolToGenAISchema(tool agent.ToolDefinition) (*genai.Schema, error)

ToolToGenAISchema converts a Tool to a genai.Schema.

func TypeToGenAISchema

func TypeToGenAISchema(t reflect.Type) *genai.Schema

TypeToGenAISchema converts a reflect.Type to a genai.Schema.

Types

type Client

type Client struct {
	// contains filtered or unexported fields
}

Client wraps the Gemini client to implement agent.LLMClient

func New

func New(client *genai.Client, modelName string) *Client

New creates a new Client

func (*Client) CreateMessages

func (c *Client) CreateMessages(ctx context.Context, req agent.MessagesRequest, onUpdate func(agent.MessagesResponse)) (agent.MessagesResponse, error)