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

fix(knext): fix nn4k error (#63)

Browse Source
This commit is contained in:
Qu 2024-01-08 11:13:49 +08:00 committed by GitHub
parent 03d19c4053
commit d6fe8ee469
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 2 additions and 83 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/knext/knext/component/builder/extractor.py
View File

@ -66,7 +66,7 @@ class LLMBasedExtractor(SPGExtractor):
def to_rest(self):
"""Transforms `LLMBasedExtractor` to REST model `ExtractNodeConfig`."""
params = dict()
params["model_config"] = json.dumps(self.llm._nn_config)
params["model_config"] = json.dumps(self.llm.init_args)
params["prompt_config"] = json.dumps(
[OperatorClient().serialize(op.to_rest()) for op in self.prompt_ops]
)

81
python/knext/knext/operator/builtin/auto_prompt.py
View File

@ -68,87 +68,6 @@ input:${input}
return self.template.replace("${input}", variables.get("input"))
def parse_response(self, response: str) -> List[SPGRecord]:
response = """
{
"spo": [
{
"subject": "甲状腺结节",
"predicate": "并发症",
"object": "甲状腺癌"
},
{
"subject": "甲状腺结节",
"predicate": "常见症状",
"object": "颈部疼痛"
},
{
"subject": "甲状腺结节",
"predicate": "常见症状",
"object": "咽喉部异物感"
},
{
"subject": "甲状腺结节",
"predicate": "常见症状",
"object": "压迫感"
},
{
"subject": "甲状腺结节",
"predicate": "适用药品",
"object": "放射性碘治疗"
},
{
"subject": "甲状腺结节",
"predicate": "适用药品",
"object": "复方碘口服液(Lugol液)"
},
{
"subject": "甲状腺结节",
"predicate": "适用药品",
"object": "丙基硫氧嘧啶(PTU)"
},
{
"subject": "甲状腺结节",
"predicate": "适用药品",
"object": "甲基硫氧嘧啶(MTU)"
},
{
"subject": "甲状腺结节",
"predicate": "适用药品",
"object": "甲硫咪唑"
},
{
"subject": "甲状腺结节",
"predicate": "适用药品",
"object": "卡比马唑"
},
{
"subject": "甲状腺结节",
"predicate": "就诊科室",
"object": "普外科"
},
{
"subject": "甲状腺结节",
"predicate": "就诊科室",
"object": "甲状腺外科"
},
{
"subject": "甲状腺结节",
"predicate": "异常指征",
"object": "血压上升"
},
{
"subject": "甲状腺结节",
"predicate": "就诊科室",
"object": "头颈外科"
},
{
"subject": "甲状腺结节",
"predicate": "发病部位",
"object": "甲状腺"
}
]
}
"""
if isinstance(response, list) and len(response) > 0:
response = response[0]
re_obj = json.loads(response)

2
python/knext/knext/operator/op.py
View File

@ -165,7 +165,7 @@ class PromptOp(BaseOp, ABC):
) -> List[Dict[str, str]]:
if isinstance(response, list) and len(response) > 0:
response = response[0]
variables.update({f"{self.name}": response})
variables.update({f"{self.__class__.__name__}": response})
return [variables]