Skip to content

Commit f50175a

Browse files
committed
refactor: Update logging messages in LLM providers to specify the model used during script generation
1 parent 8c7c37f commit f50175a

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

internal/llm/claude.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ func NewClaudeProvider(config ClaudeConfig) (*ClaudeProvider, error) {
2828
// GenerateScripts creates a main script and test script from a natural language description
2929
func (p *ClaudeProvider) GenerateScripts(ctx context.Context, description string) (ScriptPair, error) {
3030
// First generate the main script
31-
log.Info("Generating main script...")
31+
log.Info("Generating main script with Claude...")
3232
mainPrompt := p.formatPrompt(FeatureScriptPrompt, description)
3333
mainScript, err := p.generate(ctx, mainPrompt)
3434
if err != nil {
@@ -38,7 +38,7 @@ func (p *ClaudeProvider) GenerateScripts(ctx context.Context, description string
3838
log.Debug("Main script generated:\n%s", mainScript)
3939

4040
// Then generate the test script
41-
log.Info("Generating test script...")
41+
log.Info("Generating test script with Claude...")
4242
testPrompt := p.formatPrompt(TestScriptPrompt, mainScript, description)
4343
testScript, err := p.generate(ctx, testPrompt)
4444
if err != nil {

internal/llm/ollama.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ func NewOllamaProvider(config OllamaConfig) (*OllamaProvider, error) {
2828
// GenerateScripts creates a main script and test script from a natural language description
2929
func (p *OllamaProvider) GenerateScripts(ctx context.Context, description string) (ScriptPair, error) {
3030
// First generate the main script
31-
log.Info("Generating main script...")
31+
log.Info("Generating main script with Ollama...")
3232
mainPrompt := p.formatPrompt(FeatureScriptPrompt, description)
3333
mainScript, err := p.generate(ctx, mainPrompt)
3434
if err != nil {
@@ -38,7 +38,7 @@ func (p *OllamaProvider) GenerateScripts(ctx context.Context, description string
3838
log.Debug("Main script generated:\n%s", mainScript)
3939

4040
// Then generate the test script
41-
log.Info("Generating test script...")
41+
log.Info("Generating test script with Ollama...")
4242
testPrompt := p.formatPrompt(TestScriptPrompt, mainScript, description)
4343
testScript, err := p.generate(ctx, testPrompt)
4444
if err != nil {

internal/llm/openai.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ func NewOpenAIProvider(config OpenAIConfig) (*OpenAIProvider, error) {
2828
// GenerateScripts creates a main script and test script from a natural language description
2929
func (p *OpenAIProvider) GenerateScripts(ctx context.Context, description string) (ScriptPair, error) {
3030
// First generate the main script
31-
log.Info("Generating main script...")
31+
log.Info("Generating main script with OpenAI...")
3232
mainPrompt := p.formatPrompt(FeatureScriptPrompt, description)
3333
mainScript, err := p.generate(ctx, mainPrompt)
3434
if err != nil {
@@ -38,7 +38,7 @@ func (p *OpenAIProvider) GenerateScripts(ctx context.Context, description string
3838
log.Debug("Main script generated:\n%s", mainScript)
3939

4040
// Then generate the test script
41-
log.Info("Generating test script...")
41+
log.Info("Generating test script with OpenAI...")
4242
testPrompt := p.formatPrompt(TestScriptPrompt, mainScript, description)
4343
testScript, err := p.generate(ctx, testPrompt)
4444
if err != nil {

0 commit comments

Comments
 (0)