Skip to content

Commit

Permalink
feat(bridge): spg server bridge (#283)
Browse files Browse the repository at this point in the history
* x

* bridge add solver

* x
  • Loading branch information
zhuzhongshu123 authored Jan 16, 2025
1 parent 653feab commit 975ab91
Showing 1 changed file with 18 additions and 0 deletions.
18 changes: 18 additions & 0 deletions kag/bridge/spg_server_bridge.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
)

0 comments on commit 975ab91

Please sign in to comment.