ai

package

v0.1.7 Latest Latest Go to latest Published: Jan 24, 2025 License: MPL-2.0 Imports: 13 Imported by: 0

Details

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

Repository

github.com/duaraghav8/dockershrink

Links

Open Source Insights

Documentation ¶

Index ¶

Constants
func GenerateSchema[T any]() interface{}
type AIService
- func NewAIService(logger *log.Logger, client *openai.Client) *AIService
- func (ai *AIService) GenerateDockerfile(req *GenerateRequest) (string, error)
- func (ai *AIService) OptimizeDockerfile(req *OptimizeRequest) (*OptimizeResponse, error)
type GenerateRequest
type GenerateResponse
type OptimizeRequest
type OptimizeResponse

Constants ¶

View Source

const (
	// 2024_08 version is performing better than 2024_11 for dockershrink
	OpenAIPreferredModel = openai.ChatModelGPT4o2024_08_06
	MaxLLMCalls          = 5
)

View Source

const (
	ToolReadFiles         = "read_files"
	ToolDeveloperFeedback = "developer_feedback"
)

View Source

const GenerateRequestSystemPrompt = `` /* 4391-byte string literal not displayed */

View Source

const GenerateRequestUserPrompt = `` /* 170-byte string literal not displayed */

TODO: extract nodejs version from package*.json and supply it in the "generate" user prompt

View Source

const InvalidDockerfileInResponsePrompt = `` /* 148-byte string literal not displayed */

View Source

const OptimizeRequestSystemPrompt = `` /* 8845-byte string literal not displayed */

View Source

const OptimizeRequestUserPrompt = `` /* 247-byte string literal not displayed */

View Source

const RequestedFileNotFoundPrompt = `` /* 128-byte string literal not displayed */

View Source

const RuleMultistageBuildsPrompt = `` /* 4632-byte string literal not displayed */

View Source

const ToolReadFilesNoFilesSpecifiedPrompt = "No files were specified for the function, so I have nothing to return to you."

View Source

const ToolReadFilesResponseSingleFilePrompt = `{{ .Filepath }}
{{ .TripleBackticks }}
{{ .Content }}
{{ .TripleBackticks }}

`

Variables ¶

This section is empty.

Functions ¶

func GenerateSchema ¶

func GenerateSchema[T any]() interface{}

Types ¶

type AIService ¶

type AIService struct {
	L *log.Logger
	// contains filtered or unexported fields
}

func NewAIService ¶

func NewAIService(logger *log.Logger, client *openai.Client) *AIService

func (*AIService) GenerateDockerfile ¶

func (ai *AIService) GenerateDockerfile(req *GenerateRequest) (string, error)

func (*AIService) OptimizeDockerfile ¶

func (ai *AIService) OptimizeDockerfile(req *OptimizeRequest) (*OptimizeResponse, error)

OptimizeDockerfile optimizes the given Dockerfile using OpenAI GPT-4o It returns the optimized Dockerfile along with the actions taken and recommendations for further optimization.

type GenerateRequest ¶

type GenerateRequest struct {
	PackageJSON      string
	ProjectDirectory *restrictedfilesystem.RestrictedFilesystem
}

type GenerateResponse ¶

type GenerateResponse struct {
	Dockerfile string `json:"dockerfile" jsonschema_description:"The generated dockerfile"`
	Comments   string `json:"comments" jsonschema_description:"Additional comments"`
}

type OptimizeRequest ¶

type OptimizeRequest struct {
	Dockerfile   string
	Dockerignore string
	PackageJSON  string

	DockerfileStageCount uint
	ProjectDirectory     *restrictedfilesystem.RestrictedFilesystem
}

type OptimizeResponse ¶

type OptimizeResponse struct {
	Dockerfile string `json:"dockerfile" jsonschema_description:"The optimized Dockerfile"`

	Recommendations []*models.OptimizationAction `json:"recommendations" jsonschema_description:"List of Recommendations for further the Dockerfile or whole project"`
	ActionsTaken    []*models.OptimizationAction `json:"actions_taken" jsonschema_description:"List of modifictions made in the Dockerfile"`
}

Source Files ¶

View all Source files

Directories ¶

Path	Synopsis
promptcreator

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