ai

package

v0.0.2 Latest Latest Go to latest Published: Apr 9, 2025 License: MIT Imports: 6 Imported by: 0

Details

Valid go.mod file
Redistributable license
Tagged version
Stable version
Learn more about best practices

Repository

github.com/Codexiaoyi/ai-git

Links

Open Source Insights

Documentation ¶

Index ¶

func GenerateCommitMessage(prompt string, config Config) (string, error)
type AnthropicConfig
type AnthropicRequest
type AnthropicResponse
type Config
- func LoadConfig() (*Config, error)
type Message
type ModelType
type OllamaConfig
type OllamaRequest
type OllamaResponse
type OpenAIConfig
type OpenAIRequest
type OpenAIResponse

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

func GenerateCommitMessage ¶

func GenerateCommitMessage(prompt string, config Config) (string, error)

GenerateCommitMessage generates a commit message using the configured AI model

Types ¶

type AnthropicConfig ¶

type AnthropicConfig struct {
	APIKey string `yaml:"api_key" json:"api_key"`
	Model  string `yaml:"model" json:"model"`
}

AnthropicConfig holds Anthropic-specific configuration

type AnthropicRequest ¶

type AnthropicRequest struct {
	Model    string    `json:"model"`
	Messages []Message `json:"messages"`
}

AnthropicRequest represents the request structure for Anthropic API

type AnthropicResponse ¶

type AnthropicResponse struct {
	Content []Message `json:"content"`
}

AnthropicResponse represents the response structure from Anthropic API

type Config ¶

type Config struct {
	Type      ModelType       `yaml:"type" json:"type"`
	OpenAI    OpenAIConfig    `yaml:"openai,omitempty" json:"openai,omitempty"`
	Ollama    OllamaConfig    `yaml:"ollama,omitempty" json:"ollama,omitempty"`
	Anthropic AnthropicConfig `yaml:"anthropic,omitempty" json:"anthropic,omitempty"`
}

Config holds the configuration for AI models

func LoadConfig ¶

func LoadConfig() (*Config, error)

LoadConfig loads the configuration from the specified file

type Message ¶

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

Message represents a chat message

type ModelType ¶

type ModelType string

ModelType represents the type of AI model

const (
	ModelOpenAI    ModelType = "openai"
	ModelOllama    ModelType = "ollama"
	ModelAnthropic ModelType = "anthropic"
)

type OllamaConfig ¶

type OllamaConfig struct {
	BaseURL string `yaml:"base_url" json:"base_url"`
	Model   string `yaml:"model" json:"model"`
}

OllamaConfig holds Ollama-specific configuration

type OllamaRequest ¶

type OllamaRequest struct {
	Model    string    `json:"model"`
	Messages []Message `json:"messages"`
	Stream   bool      `json:"stream"` // Always set to false
}

OllamaRequest represents the request structure for Ollama API

type OllamaResponse ¶

type OllamaResponse struct {
	Message Message `json:"message"`
}

OllamaResponse represents the response structure from Ollama API

type OpenAIConfig ¶

type OpenAIConfig struct {
	APIKey string `yaml:"api_key" json:"api_key"`
	Model  string `yaml:"model" json:"model"`
}

OpenAIConfig holds OpenAI-specific configuration

type OpenAIRequest ¶

type OpenAIRequest struct {
	Model    string    `json:"model"`
	Messages []Message `json:"messages"`
	Stream   bool      `json:"stream"` // Always set to false
}

OpenAIRequest represents the request structure for OpenAI API

type OpenAIResponse ¶

type OpenAIResponse struct {
	Choices []struct {
		Message Message `json:"message"`
	} `json:"choices"`
}

OpenAIResponse represents the response structure from OpenAI API

Source Files ¶

View all Source files

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL