try 2.4

api/chain.go

@@ -31,10 +31,10 @@ func InvokeChain(apiKey string, userQuery string) (ChatOutput, error) {
 	prompt := string(systemPrompt) + "\n\nUser Query: " + userQuery
 
 	// Write prompt to log file ==> To be removed in production
-	err = writePromptToLogFile(prompt)
-	if err != nil {
-		return ChatOutput{}, fmt.Errorf("error writing prompt to log file: %v", err)
-	}
+	// err = writePromptToLogFile(prompt)
+	// if err != nil {
+	// 	return ChatOutput{}, fmt.Errorf("error writing prompt to log file: %v", err)
+	// }
 
 	ctx := context.Background()
 
@@ -47,6 +47,8 @@ func InvokeChain(apiKey string, userQuery string) (ChatOutput, error) {
 		llms.WithJSONMode(),
 	)
 
+	log.Println("Response Content: ", responseContent)
+
 	if err != nil {
 		return ChatOutput{}, fmt.Errorf("error in generating content: %v", err)
 	}
@@ -71,19 +73,19 @@ func extractResponse(responseContent string) string {
 	return responseContent
 }
 
-func writePromptToLogFile(prompt string) error {
-	// if _, err := os.Stat("./log"); os.IsNotExist(err) {
-	// 	os.Mkdir("./log", 0755)
-	// }
-	file, err := os.OpenFile("./prompt.log", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
-	if err != nil {
-		return fmt.Errorf("error opening log file: %v", err)
-	}
-	defer file.Close()
-
-	if _, err := file.WriteString(prompt); err != nil {
-		return fmt.Errorf("error writing to log file: %v", err)
-	}
-
-	return nil
-}
+// func writePromptToLogFile(prompt string) error {
+// 	// if _, err := os.Stat("./log"); os.IsNotExist(err) {
+// 	// 	os.Mkdir("./log", 0755)
+// 	// }
+// 	file, err := os.OpenFile("./prompt.log", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
+// 	if err != nil {
+// 		return fmt.Errorf("error opening log file: %v", err)
+// 	}
+// 	defer file.Close()
+
+// 	if _, err := file.WriteString(prompt); err != nil {
+// 		return fmt.Errorf("error writing to log file: %v", err)
+// 	}
+
+// 	return nil
+// }