Update tests for issue 1229 (#1231)

Signed-off-by: Mustafa <mustafa.cetin@intel.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2024-12-06 17:07:52 -08:00
parent bde285dfce
commit 07e47a1f38
3 changed files with 9 additions and 7 deletions
--- a/DocSum/docsum.py
+++ b/DocSum/docsum.py
@@ -52,7 +52,9 @@ class DocSumService(Gateway):
            use_remote_service=True,
            service_type=ServiceType.LLM,
        )
-        self.megaservice.add(llm)
+
+        self.megaservice.add(data).add(llm)
+        self.megaservice.flow_to(data, llm)

    async def handle_request(self, request: Request, files: List[UploadFile] = File(default=None)):