fix: escape characters escaping

kqvanity · kqvanity · commit f0ee1484d66f · 2024-10-16T10:23:58.000+03:00
diff --git a/main.go b/main.go
@@ -1,5 +1,5 @@
 package main
 
 func main() {
-
+	MongoRestClient()
 }
diff --git a/mongo.go b/mongo.go
@@ -250,12 +250,8 @@ func (s *RestStreamCallback) OnReferences(r ReferencesMsg) {
 	s.f.Flush()
 }
 
-// FIXME: ai doesn't parse the output quite nicely with escape chars and whatnot
 func escapeJSON(s string) string {
-	//if !strings.Contains(s, " ") {
-	//	//fmt.Println("contains", s)
-	//	return s
-	//}
+	s = strings.ReplaceAll(s, "\\n", "\n")
 	b, _ := json.Marshal(s)
 	return string(b[1 : len(b)-1])
 }
@@ -305,6 +301,7 @@ func MongoRestClient() {
 		log.Fatal(err)
 	}
 }
+
 func cHandler(c *gin.Context) {
 	var req OpenaiRequest
 	flusher, ok := c.Writer.(http.Flusher)

Original file line number	Diff line number	Diff line change
`@@ -250,12 +250,8 @@ func (s *RestStreamCallback) OnReferences(r ReferencesMsg) {`
`250`	`250`	`s.f.Flush()`
`251`	`251`	`}`
`252`	`252`
`253`		`-// FIXME: ai doesn't parse the output quite nicely with escape chars and whatnot`
`254`	`253`	`func escapeJSON(s string) string {`
`255`		`- //if !strings.Contains(s, " ") {`
`256`		`- // //fmt.Println("contains", s)`
`257`		`- // return s`
`258`		`- //}`
	`254`	`+ s = strings.ReplaceAll(s, "\\n", "\n")`
`259`	`255`	`b, _ := json.Marshal(s)`
`260`	`256`	`return string(b[1 : len(b)-1])`
`261`	`257`	`}`
`@@ -305,6 +301,7 @@ func MongoRestClient() {`
`305`	`301`	`log.Fatal(err)`
`306`	`302`	`}`
`307`	`303`	`}`
	`304`	`+`
`308`	`305`	`func cHandler(c *gin.Context) {`
`309`	`306`	`var req OpenaiRequest`
`310`	`307`	`flusher, ok := c.Writer.(http.Flusher)`