prebuilt

Documentation

Index

• func NewCodeRefactorAgent(engine engines.LLM) agents.Agent[CodeBaseRefactorRequest, CodeBaseRefactorResponse]
• func NewGitAssistantAgent(engine engines.LLM, ...) agents.Agent[GitAssistantRequest, GitAssistantResponse]
• func NewTradeAssistantAgent(engine engines.LLM, wolframAlphaAppID string) agents.Agent[TradeAssistantRequest, TradeAssistantResponse]
• func NewUnitTestWriter(engine engines.LLM, codeValidator func(code string) error) (agents.Agent[UnitTestWriterRequest, UnitTestWriterResponse], error)
• type CodeBaseRefactorRequest
  • func (req CodeBaseRefactorRequest) Encode() string
  • func (req CodeBaseRefactorRequest) Schema() string
• type CodeBaseRefactorResponse
  • func (resp CodeBaseRefactorResponse) Encode() string
  • func (resp CodeBaseRefactorResponse) Schema() string
• type GitAssistantRequest
  • func (req GitAssistantRequest) Encode() string
  • func (req GitAssistantRequest) Schema() string
• type GitAssistantResponse
  • func (resp GitAssistantResponse) Encode() string
  • func (resp GitAssistantResponse) Schema() string
• type Recommendation
• type TradeAssistantRequest
  • func (r TradeAssistantRequest) Encode() string
  • func (r TradeAssistantRequest) Schema() string
• type TradeAssistantResponse
  • func (r TradeAssistantResponse) Encode() string
  • func (r TradeAssistantResponse) Schema() string
• type UnitTestWriterRequest
  • func (r UnitTestWriterRequest) Encode() string
  • func (r UnitTestWriterRequest) Schema() string
• type UnitTestWriterResponse
  • func (r UnitTestWriterResponse) Encode() string
  • func (r UnitTestWriterResponse) Schema() string

Functions

func NewCodeRefactorAgent

func NewCodeRefactorAgent(engine engines.LLM) agents.Agent[CodeBaseRefactorRequest, CodeBaseRefactorResponse]

func NewGitAssistantAgent

func NewGitAssistantAgent(engine engines.LLM, actionConfirmationHook func(action *agents.ChainAgentAction) bool, additionalTools ...tools.Tool) agents.Agent[GitAssistantRequest, GitAssistantResponse]

func NewTradeAssistantAgent

func NewTradeAssistantAgent(engine engines.LLM, wolframAlphaAppID string) agents.Agent[TradeAssistantRequest, TradeAssistantResponse]

func NewUnitTestWriter

func NewUnitTestWriter(engine engines.LLM, codeValidator func(code string) error) (agents.Agent[UnitTestWriterRequest, UnitTestWriterResponse], error)

Types

type CodeBaseRefactorRequest

type CodeBaseRefactorRequest struct {
	Dir  string
	Goal string
}

func (CodeBaseRefactorRequest) Encode

func (req CodeBaseRefactorRequest) Encode() string

func (CodeBaseRefactorRequest) Schema

func (req CodeBaseRefactorRequest) Schema() string

type CodeBaseRefactorResponse

type CodeBaseRefactorResponse struct {
	RefactoredFiles map[string]string `json:"refactored_files"`
}

func (CodeBaseRefactorResponse) Encode

func (resp CodeBaseRefactorResponse) Encode() string

func (CodeBaseRefactorResponse) Schema

func (resp CodeBaseRefactorResponse) Schema() string

type GitAssistantRequest

type GitAssistantRequest struct {
	Instruction string
	GitStatus   string
	CurrentDate string
}

func (GitAssistantRequest) Encode

func (req GitAssistantRequest) Encode() string

func (GitAssistantRequest) Schema

func (req GitAssistantRequest) Schema() string

type GitAssistantResponse

type GitAssistantResponse struct {
	Summary string `json:"summary"`
}

func (GitAssistantResponse) Encode

func (resp GitAssistantResponse) Encode() string

func (GitAssistantResponse) Schema

func (resp GitAssistantResponse) Schema() string

type Recommendation

type Recommendation string

const (
	RecommendationBuy  Recommendation = "buy"
	RecommendationSell Recommendation = "sell"
	RecommendationHold Recommendation = "hold"
)

type TradeAssistantRequest

type TradeAssistantRequest struct {
	Stocks []string `json:"stocks"`
}

func (TradeAssistantRequest) Encode

func (r TradeAssistantRequest) Encode() string

func (TradeAssistantRequest) Schema

func (r TradeAssistantRequest) Schema() string

type TradeAssistantResponse

type TradeAssistantResponse struct {
	Recommendations map[string]Recommendation
}

func (TradeAssistantResponse) Encode

func (r TradeAssistantResponse) Encode() string

func (TradeAssistantResponse) Schema

func (r TradeAssistantResponse) Schema() string

type UnitTestWriterRequest

type UnitTestWriterRequest struct {
	SourceFile  string `json:"source_file"`
	ExampleFile string `json:"example_file"`
}

func (UnitTestWriterRequest) Encode

func (r UnitTestWriterRequest) Encode() string

func (UnitTestWriterRequest) Schema

func (r UnitTestWriterRequest) Schema() string

type UnitTestWriterResponse

type UnitTestWriterResponse struct {
	UnitTestFile string `json:"unit_test_file"`
}

func (UnitTestWriterResponse) Encode

func (r UnitTestWriterResponse) Encode() string

func (UnitTestWriterResponse) Schema

func (r UnitTestWriterResponse) Schema() string