pr example

Signed-off-by: Future Outlier <[email protected]>
flyteorg · Dec 4, 2023 · a670fd2 · a670fd2
1 parent 94e6998
commit a670fd2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/flytekit/clis/sdk_in_container/serve.py b/flytekit/clis/sdk_in_container/serve.py
@@ -45,12 +45,13 @@ def serve(_: click.Context, port, worker, timeout):
 
 async def _start_grpc_server(port: int, worker: int, timeout: int):
     click.secho("Starting up the server to expose the prometheus metrics...", fg="blue")
+    print("port:", port)
     from flytekit.extend.backend.agent_service import AsyncAgentService, AgentMetadataService
 
     try:
         from prometheus_client import start_http_server
 
-        start_http_server(9090)
+        start_http_server(10000 + port)
     except ImportError as e:
         click.secho(f"Failed to start the prometheus server with error {e}", fg="red")
     click.secho("Starting the agent service...", fg="blue")

diff --git a/flytekit/extend/backend/agent_service.py b/flytekit/extend/backend/agent_service.py
@@ -147,6 +147,7 @@ async def GetAgent(self, request: GetAgentRequest, context: grpc.ServicerContext
         )
 
     async def ListAgent(self, request: ListAgentsRequest, context: grpc.ServicerContext) -> ListAgentsResponse:
+        print("@@@ ListAgent is called!")
         agents = []
         for name in AgentRegistry._REGISTRY.keys():
             agents.append(