vertex

README

Vertex Gemini Provider Generator

This directory contains a Python script for generating the Vertex provider code based on the Studio provider implementation.

Usage

To generate the Vertex provider code, run the following command from the root of the project:

python3 provider/vertex/vertex_generator.py --srcs provider/studio/studio.go --outs provider/vertex/vertex.go

Documentation

Index

• type Endpoint
  • func NewEndpoint(projectId string, region string) (*Endpoint, error)
  • func (ep *Endpoint) GenerateChatCompletion(ctx context.Context, openaiRequest *openai.ChatCompletionRequest) (*openai.ChatCompletionResponse, error)
  • func (ep *Endpoint) Ping(ctx context.Context) (time.Duration, error)
  • func (ep *Endpoint) Provider() string
  • func (ep *Endpoint) Region() string
  • func (ep *Endpoint) Shutdown() error

Types

type Endpoint

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

func NewEndpoint

func NewEndpoint(projectId string, region string) (*Endpoint, error)

func (*Endpoint) GenerateChatCompletion

func (ep *Endpoint) GenerateChatCompletion(ctx context.Context, openaiRequest *openai.ChatCompletionRequest) (*openai.ChatCompletionResponse, error)

func (*Endpoint) Ping

func (ep *Endpoint) Ping(ctx context.Context) (time.Duration, error)

func (*Endpoint) Provider

func (ep *Endpoint) Provider() string

func (*Endpoint) Region

func (ep *Endpoint) Region() string

func (*Endpoint) Shutdown

func (ep *Endpoint) Shutdown() error