fix: api

KusionStack · Aug 25, 2024 · c2085c0 · c2085c0
1 parent 1c7ae79
commit c2085c0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 14 deletions.
diff --git a/pkg/infra/ai/azureopenai.go b/pkg/infra/ai/azureopenai.go
@@ -47,16 +47,10 @@ func (c *AzureAIClient) Configure(cfg AIConfig) error {
 	return nil
 }
 
-func (c *AzureAIClient) Generate(ctx context.Context, prompt string, serviceType string) (string, error) {
-	servicePrompt := ServicePromptMap[serviceType]
-
+func (c *AzureAIClient) Generate(ctx context.Context, prompt string) (string, error) {
 	resp, err := c.client.CreateChatCompletion(ctx, openai.ChatCompletionRequest{
 		Model: c.model,
 		Messages: []openai.ChatCompletionMessage{
-			{
-				Role:    openai.ChatMessageRoleSystem,
-				Content: servicePrompt,
-			},
 			{
 				Role:    openai.ChatMessageRoleUser,
 				Content: prompt,

diff --git a/pkg/infra/ai/openai.go b/pkg/infra/ai/openai.go
@@ -48,16 +48,10 @@ func (c *OpenAIClient) Configure(cfg AIConfig) error {
 	return nil
 }
 
-func (c *OpenAIClient) Generate(ctx context.Context, prompt string, serviceType string) (string, error) {
-	servicePrompt := ServicePromptMap[serviceType]
-
+func (c *OpenAIClient) Generate(ctx context.Context, prompt string) (string, error) {
 	resp, err := c.client.CreateChatCompletion(ctx, openai.ChatCompletionRequest{
 		Model: c.model,
 		Messages: []openai.ChatCompletionMessage{
-			{
-				Role:    openai.ChatMessageRoleSystem,
-				Content: servicePrompt,
-			},
 			{
 				Role:    openai.ChatMessageRoleUser,
 				Content: prompt,