From 975ab912e2338e92234528df2263bc6957c70882 Mon Sep 17 00:00:00 2001
From: zhuzhongshu123 <152354526+zhuzhongshu123@users.noreply.github.com>
Date: Thu, 16 Jan 2025 10:45:08 +0800
Subject: [PATCH] feat(bridge): spg server bridge (#283)

* x

* bridge add solver

* x
---
 kag/bridge/spg_server_bridge.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/kag/bridge/spg_server_bridge.py b/kag/bridge/spg_server_bridge.py
index 48d53c44..c192bf3e 100644
--- a/kag/bridge/spg_server_bridge.py
+++ b/kag/bridge/spg_server_bridge.py
@@ -58,3 +58,21 @@ def run_vectorizer_config_check(self, vec_config):
         )
 
         return VectorizeModelConfigChecker().check(vec_config)
+
+    def run_solver(
+        self,
+        project_id,
+        task_id,
+        query,
+        is_report=True,
+        host_addr="http://127.0.0.1:8887",
+    ):
+        from kag.solver.main_solver import SolverMain
+
+        return SolverMain().invoke(
+            project_id=project_id,
+            task_id=task_id,
+            query=query,
+            is_report=is_report,
+            host_addr=host_addr,
+        )
\ No newline at end of file