fix ansysgpt no output (#97)

ansys · Nov 7, 2024 · 43e336e · 43e336e
1 parent da717a8
commit 43e336e
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 18 deletions.
diff --git a/pkg/externalfunctions/ansysgpt.go b/pkg/externalfunctions/ansysgpt.go
@@ -652,8 +652,6 @@ func AisAcsSemanticHybridSearchs(
 		output = append(output, results...)
 	}
 
-	fmt.Println(len(output))
-
 	return output
 }
 
@@ -738,20 +736,23 @@ func AisPerformLLMFinalRequest(systemTemplate string,
 	}
 
 	// create json string from context
-	contextString := "{"
-	chunkNr := 1
-	for _, example := range context {
-		json, err := json.Marshal(example)
-		if err != nil {
-			logging.Log.Errorf(internalstates.Ctx, "Error marshalling context: %v", err)
-			return "", nil
+	contextString := ""
+	if len(context) != 0 {
+		contextString += "{"
+		chunkNr := 1
+		for _, example := range context {
+			json, err := json.Marshal(example)
+			if err != nil {
+				logging.Log.Errorf(internalstates.Ctx, "Error marshalling context: %v", err)
+				return "", nil
+			}
+			contextString += fmt.Sprintf("\"chunk %v\": %v", chunkNr, string(json)) + ", "
+			chunkNr++
 		}
-		contextString += fmt.Sprintf("\"chunk %v\": %v", chunkNr, string(json)) + ", "
-		chunkNr++
+		// remove last comma, then add closing bracket
+		contextString = contextString[:len(contextString)-2]
+		contextString += "}"
 	}
-	// remove last comma, then add closing bracket
-	contextString = contextString[:len(contextString)-2]
-	contextString += "}"
 
 	// create string from prohibited words
 	prohibitedWordsString := ""

diff --git a/pkg/grpcserver/grpcserver.go b/pkg/grpcserver/grpcserver.go
@@ -145,15 +145,15 @@ func (s *server) RunFunction(ctx context.Context, req *allieflowkitgrpc.Function
 		var err error
 		inputs[i], err = typeconverters.ConvertStringToGivenType(input.Value, functionDefinition.Input[i].GoType)
 		if err != nil {
-			return nil, fmt.Errorf("error converting input %s to type %s: %v", input.Name, functionDefinition.Input[i].GoType, err)
+			return nil, fmt.Errorf("error converting input '%s' of function '%s' to type '%s': %v", input.Name, req.Name, functionDefinition.Input[i].GoType, err)
 		}
 
 		// check for option sets and convert values
 		if len(functionDefinition.Input[i].Options) > 0 {
 			// convert value to correct type
 			inputs[i], err = convertOptionSetValues(functionDefinition.Name, input.Name, inputs[i])
 			if err != nil {
-				return nil, fmt.Errorf("error converting input %s to type %s: %v", input.Name, functionDefinition.Input[i].GoType, err)
+				return nil, fmt.Errorf("error converting option set input '%s' of function '%s' to type '%s': %v", input.Name, req.Name, functionDefinition.Input[i].GoType, err)
 			}
 		}
 	}
@@ -230,15 +230,15 @@ func (s *server) StreamFunction(req *allieflowkitgrpc.FunctionInputs, stream all
 		var err error
 		inputs[i], err = typeconverters.ConvertStringToGivenType(input.Value, functionDefinition.Input[i].GoType)
 		if err != nil {
-			return fmt.Errorf("error converting input %s to type %s: %v", input.Name, functionDefinition.Input[i].GoType, err)
+			return fmt.Errorf("error converting input '%s' of function '%s' to type '%s': %v", input.Name, req.Name, functionDefinition.Input[i].GoType, err)
 		}
 
 		// check for option sets and convert values
 		if len(functionDefinition.Input[i].Options) > 0 {
 			// convert value to correct type
 			inputs[i], err = convertOptionSetValues(functionDefinition.Name, input.Name, inputs[i])
 			if err != nil {
-				return fmt.Errorf("error converting input %s to type %s: %v", input.Name, functionDefinition.Input[i].GoType, err)
+				return fmt.Errorf("error converting option set input '%s' of function '%s' to type '%s': %v", input.Name, req.Name, functionDefinition.Input[i].GoType, err)
 			}
 		}
 	}