new index and 25r1 logic for ansys gpt (#110)

ansys · Jan 29, 2025 · 2cefa0b · 2cefa0b
1 parent ccc326b
commit 2cefa0b
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 7 deletions.
diff --git a/pkg/externalfunctions/ansysgpt.go b/pkg/externalfunctions/ansysgpt.go
@@ -581,29 +581,39 @@ func AisPerformLLMRephraseRequest(systemTemplate string, userTemplate string, qu
 //
 // Returns:
 //   - indexList: the index list
-func AisReturnIndexList(accessPoint string, physics []string) (indexList []string) {
+func AisReturnIndexList(accessPoint string, physics []string, version []string) (indexList []string) {
 	indexList = []string{}
 
 	switch accessPoint {
 	case "ansysgpt-general", "ais-embedded":
-		if len(physics) == 1 && physics[0] == "scade" {
+		switch {
+		case len(physics) == 1 && physics[0] == "scade":
 			// special case for Scade One
-			indexList = append(indexList, "external-product-documentation-public")
-		} else {
+		default:
 			// default ais case
 			indexList = append(indexList,
 				"granular-ansysgpt",
 				"ansysgpt-documentation-2023r2",
 				"lsdyna-documentation-r14",
 				"scade-documentation-2023r2",
 				"external-marketing",
-				"external-product-documentation-public",
 				"external-learning-hub",
 				"external-crtech-thermal-desktop",
 				"external-release-notes",
 				"external-zemax-websites",
 			)
 		}
+
+		// if version is only 25r1, add only the 25r1 index, otherwise add both
+		if len(version) == 1 && strings.Contains(strings.ToLower(version[0]), "25r1") {
+			indexList = append(indexList, "external-product-documentation-public-25r1")
+		} else {
+			indexList = append(indexList,
+				"external-product-documentation-public",
+				"external-product-documentation-public-25r1",
+			)
+		}
+
 	case "ansysgpt-scbu":
 		indexList = append(indexList,
 			"ansysgpt-scbu",

diff --git a/pkg/externalfunctions/privatefunctions.go b/pkg/externalfunctions/privatefunctions.go
@@ -896,7 +896,7 @@ func getFieldsAndReturnProperties(indexName string) (searchedEmbeddedFields stri
 	case "external-crtech-thermal-desktop":
 		searchedEmbeddedFields = "contentVector, sourceTitle_lvl1_vctr, sourceTitle_lvl2_vctr, sourceTitle_lvl3_vctr"
 		returnedProperties = "token_size, physics, typeOFasset, product, version, weight, bridge_id, content, sourceTitle_lvl2, sourceURL_lvl2, sourceTitle_lvl3, sourceURL_lvl3"
-	case "external-product-documentation-public", "external-learning-hub", "external-release-notes", "external-zemax-websites", "external-scbu-learning-hub":
+	case "external-product-documentation-public", "external-product-documentation-public-25r1", "external-learning-hub", "external-release-notes", "external-zemax-websites", "external-scbu-learning-hub":
 		searchedEmbeddedFields = "contentVector, sourceTitle_lvl1_vctr, sourceTitle_lvl2_vctr, sourceTitle_lvl3_vctr"
 		returnedProperties = "token_size, physics, typeOFasset, product, version, weight, content, sourceTitle_lvl2, sourceURL_lvl2, sourceTitle_lvl3, sourceURL_lvl3"
 	case "scbu-data-except-alh":
@@ -983,7 +983,7 @@ func extractAndConvertACSResponse(body []byte, indexName string) (output []share
 			})
 		}
 
-	case "external-product-documentation-public", "external-learning-hub", "external-crtech-thermal-desktop", "external-release-notes", "external-zemax-websites", "external-scbu-learning-hub", "scbu-data-except-alh":
+	case "external-product-documentation-public", "external-product-documentation-public-25r1", "external-learning-hub", "external-crtech-thermal-desktop", "external-release-notes", "external-zemax-websites", "external-scbu-learning-hub", "scbu-data-except-alh":
 		respObjectCrtech := ACSSearchResponseStructCrtech{}
 		err := json.Unmarshal(body, &respObjectCrtech)
 		if err != nil {