Update response formatting.

qiuosier · qiuosier · commit f2d42d4672e7 · 2023-12-02T21:06:13.000-05:00
diff --git a/ads/llm/langchain/plugins/llm_md.py b/ads/llm/langchain/plugins/llm_md.py
@@ -234,7 +234,7 @@ def _construct_json_body(self, prompt, params):
         }
 
     def _process_response(self, response_json: dict):
-        return str(response_json.get("generated_text", response_json)) + "\n"
+        return str(response_json.get("generated_text", response_json))
 
 
 class ModelDeploymentVLLM(ModelDeploymentLLM):

Original file line number	Diff line number	Diff line change
`@@ -234,7 +234,7 @@ def _construct_json_body(self, prompt, params):`
`234`	`234`	`}`
`235`	`235`
`236`	`236`	`def _process_response(self, response_json: dict):`
`237`		`- return str(response_json.get("generated_text", response_json)) + "\n"`
	`237`	`+ return str(response_json.get("generated_text", response_json))`
`238`	`238`
`239`	`239`
`240`	`240`	`class ModelDeploymentVLLM(ModelDeploymentLLM):`