Skip to content

Commit 8d7d274

Browse files
committed
fix: GetFuncCallResult bug
1 parent 786ef99 commit 8d7d274

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

apis/record/infer.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,7 @@ func GetFuncCallResult(
268268
case "python": {
269269
parameters := data["parameters"].(map[string]interface{})
270270
codeContent := parameters["code"].(string)
271-
if inputC, ok := parameters["input"]; ok == true {
271+
if inputC, ok := parameters["input"]; ok {
272272
inputContent = inputC.(string)
273273
}
274274
url = UrlConfig.pythonUrl
@@ -284,7 +284,7 @@ func GetFuncCallResult(
284284
case "retrieval": {
285285
url = UrlConfig.retrievalUrl
286286
parameters := data["parameters"].(map[string]interface{})
287-
query := parameters["query"].([]string)
287+
query := parameters["query"].(string)
288288
body = map[string]interface{} {
289289
"query": query,
290290
"top_k": 5,
@@ -295,7 +295,7 @@ func GetFuncCallResult(
295295
case "document": {
296296
url = UrlConfig.documentUrl
297297
parameters := data["parameters"].(map[string]interface{})
298-
query := parameters["question"].([]string)
298+
query := parameters["question"].(string)
299299

300300
body = map[string]interface{} {
301301
"query": query,
@@ -306,7 +306,7 @@ func GetFuncCallResult(
306306
case "bingapi": {
307307
url = UrlConfig.bingapiUrl
308308
parameters := data["parameters"].(map[string]interface{})
309-
query := parameters["question"].([]string)
309+
query := parameters["question"].(string)
310310
topK := 2
311311
if val, exists := parameters["top_k"]; exists {
312312
topK = val.(int)

0 commit comments

Comments
 (0)