update test

This commit is contained in:
usamimeri_renko 2024-04-26 17:30:03 +08:00
parent 83c8ccb6b9
commit b0ed292fa7
2 changed files with 15 additions and 5 deletions

View file

@ -74,6 +74,11 @@ class AmazonProvider(BaseBedrockProvider):
def _get_completion_from_dict(self, rsp_dict: dict) -> str:
return rsp_dict['results'][0]['outputText']
def get_choice_text_from_stream(self, event) -> str:
rsp_dict = json.loads(event["chunk"]["bytes"])
completions = rsp_dict["outputText"]
return completions
PROVIDERS = {