yujunjun/KAG
1
0
Fork 0
mirror of https://github.com/OpenSPG/KAG.git synced 2025-06-27 03:20:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(solver): vectorize_model fix generate_key (#518)

Browse Source 
* add async return

* turn vectorize model to singleton

* turn vectorize model to singleton

* add schema type linker

* fix black format

* bugfix vectorize

---------

Co-authored-by: 钟书 <zhongshu.zzs@antgroup.com>
This commit is contained in:
royzhao 2025-05-06 21:02:18 +08:00 committed by GitHub
parent 404d79b304
commit 6a4ab468fe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
kag/interface/common/vectorize_model.py
View File

@ -106,7 +106,7 @@ class VectorizeModelABC(Registrable):
raise RuntimeError(message) from ex raise RuntimeError(message) from ex
@classmethod @classmethod
def generate_key(self, cls, *args, **kwargs) -> str: def generate_key(cls, *args, **kwargs) -> str:
return f"{cls}" return f"{cls}"
@retry(stop=stop_after_attempt(3), reraise=True) @retry(stop=stop_after_attempt(3), reraise=True)

2
kag/solver/reporter/open_spg_reporter.py
View File

@ -424,7 +424,7 @@ Rewritten question:\n{content}
self.last_report = request self.last_report = request
if self.last_report.to_dict() == request.to_dict(): if self.last_report.to_dict() == request.to_dict():
return return
logger.info(f"do_report: {request} ret={ret}") logger.info(f"do_report: {content.answer} think={content.think} ret={ret}")
self.last_report = request self.last_report = request
except Exception as e: except Exception as e: