(incubator-hugegraph-ai) branch main updated: fix(llm): add missing 'properties' in gremlin prompt formatting (#298)

[jin]

This is an automated email from the ASF dual-hosted git repository.

jin pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/incubator-hugegraph-ai.git


The following commit(s) were added to refs/heads/main by this push:
     new dedf4aee fix(llm): add missing 'properties' in gremlin prompt 
formatting (#298)
dedf4aee is described below

commit dedf4aee6cdfa3d5d39a272ad4b159acc584fa11
Author: Sean <2579935...@qq.com>
AuthorDate: Mon Aug 11 19:23:42 2025 +0800

    fix(llm): add missing 'properties' in gremlin prompt formatting (#298)
---
 .../src/hugegraph_llm/operators/llm_op/gremlin_generate.py       | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git 
a/hugegraph-llm/src/hugegraph_llm/operators/llm_op/gremlin_generate.py 
b/hugegraph-llm/src/hugegraph_llm/operators/llm_op/gremlin_generate.py
index 81aef300..11f0f602 100644
--- a/hugegraph-llm/src/hugegraph_llm/operators/llm_op/gremlin_generate.py
+++ b/hugegraph-llm/src/hugegraph_llm/operators/llm_op/gremlin_generate.py
@@ -63,6 +63,11 @@ class GremlinGenerateSynthesize:
             return None
         return "\n".join([f"- '{vid}'" for vid in vertices])
 
+    def _format_properties(self, properties: Optional[List[tuple]]) -> 
Optional[str]:
+        if not properties:
+            return None
+        return str(properties)
+
     async def async_generate(self, context: Dict[str, Any]):
         async_tasks = {}
         query = context.get("query")
@@ -72,6 +77,7 @@ class GremlinGenerateSynthesize:
             schema=self.schema,
             example=self._format_examples(examples=raw_example),
             vertices=self._format_vertices(vertices=self.vertices),
+            properties=self._format_properties(properties=None),
         )
         async_tasks["raw_answer"] = 
asyncio.create_task(self.llm.agenerate(prompt=raw_prompt))
 
@@ -81,6 +87,7 @@ class GremlinGenerateSynthesize:
             schema=self.schema,
             example=self._format_examples(examples=examples),
             vertices=self._format_vertices(vertices=self.vertices),
+            properties=self._format_properties(properties=None),
         )
         async_tasks["initialized_answer"] = 
asyncio.create_task(self.llm.agenerate(prompt=init_prompt))
 
@@ -102,6 +109,7 @@ class GremlinGenerateSynthesize:
             schema=self.schema,
             example=self._format_examples(examples=raw_example),
             vertices=self._format_vertices(vertices=self.vertices),
+            properties=self._format_properties(properties=None),
         )
         raw_response = self.llm.generate(prompt=raw_prompt)
 
@@ -111,6 +119,7 @@ class GremlinGenerateSynthesize:
             schema=self.schema,
             example=self._format_examples(examples=examples),
             vertices=self._format_vertices(vertices=self.vertices),
+            properties=self._format_properties(properties=None),
         )
         initialized_response = self.llm.generate(prompt=init_prompt)