feat: add llm wenxinyiyan & config util & spo_triple_extract #27
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters
dosubot bot added size:L dosubot bot added size:XL dosubot bot added size:XXL 
simon824 changed the title 
Member Author
| @imbajin @lzyxx77 @liuxiaocs7 PTAL, thanks! |
liuxiaocs7 reviewed Jan 22, 2024
Member
There was a problem hiding this comment.
Overall looks good to me, left some minor comments
wenxinyiyancould be replaced withERNIEorERNIE-Bot?
because: https://cloud.baidu.com/doc/WENXINWORKSHOP/s/Nlks5zkzu and more international (at this point CC: @imbajin)- There are many automatic formatting modifications in this pr. could we add corresponding commands in the readme or other locations to stay code style unified.
hugegraph-llm/src/hugegraph_llm/operators/kg_construction_task.py Outdated Show resolved Hide resolved
simon824 commented Jan 23, 2024
| we need to manually fix all the warnings mentioned below before commit! " | ||
| export PYTHONPATH=${ROOT_DIR}/hugegraph-llm/src:${ROOT_DIR}/hugegraph-python-client/src | ||
| pylint --rcfile=${ROOT_DIR}/style/pylint.conf ${ROOT_DIR}/hugegraph-llm | ||
| #pylint --rcfile=${ROOT_DIR}/style/pylint.conf ${ROOT_DIR}/hugegraph-python-client |
Member Author
There was a problem hiding this comment.
will fix hugegraph-python-client code style in next pr
liuxiaocs7 reviewed Jan 23, 2024
hugegraph-llm/src/hugegraph_llm/operators/llm_op/info_extract.py Outdated Show resolved Hide resolved
hugegraph-llm/src/hugegraph_llm/operators/llm_op/info_extract.py Outdated Show resolved Hide resolved
hugegraph-llm/src/hugegraph_llm/operators/llm_op/info_extract.py Outdated Show resolved Hide resolved
hugegraph-llm/src/hugegraph_llm/operators/llm_op/info_extract.py Outdated Show resolved Hide resolved
javeme reviewed Jan 23, 2024
hugegraph-llm/src/hugegraph_llm/operators/common_op/print_result.py Outdated Show resolved Hide resolved
hugegraph-llm/src/hugegraph_llm/operators/hugegraph_op/commit_to_hugegraph.py Outdated Show resolved Hide resolved
imbajin reviewed Jan 24, 2024
Comment on lines +31 to +52
| return """You are a data scientist working for a company that is building a graph database. | ||
| Your task is to extract information from data and convert it into a graph database. Provide a | ||
| set of Nodes in the form [ENTITY_ID, TYPE, PROPERTIES] and a set of relationships in the form | ||
| [ENTITY_ID_1, RELATIONSHIP, ENTITY_ID_2, PROPERTIES] and a set of NodesSchemas in the form [ | ||
| ENTITY_TYPE, PRIMARY_KEY, PROPERTIES] and a set of RelationshipsSchemas in the form [ | ||
| ENTITY_TYPE_1, RELATIONSHIP, ENTITY_TYPE_2, PROPERTIES] It is important that the ENTITY_ID_1 | ||
| and ENTITY_ID_2 exists as nodes with a matching ENTITY_ID. If you can't pair a relationship | ||
| with a pair of nodes don't add it. When you find a node or relationship you want to add try | ||
| to create a generic TYPE for it that describes the entity you can also think of it as a label. | ||
| Here is an example The input you will be given: Data: Alice lawyer and is 25 years old and Bob | ||
| is her roommate since 2001. Bob works as a journalist. Alice owns a the webpage www.alice.com | ||
| and Bob owns the webpage www.bob.com. The output you need to provide: Nodes: ["Alice", "Person", | ||
| {"age": 25, "occupation": "lawyer", "name": "Alice"}], ["Bob", "Person", {"occupation": | ||
| "journalist", "name": "Bob"}], ["alice.com", "Webpage", {"name": "alice.com", | ||
| "url": "www.alice.com"}], ["bob.com", "Webpage", {"name": "bob.com", "url": "www.bob.com"}] | ||
| Relationships: [{"Person": "Alice"}, "roommate", {"Person": "Bob"}, {"start": 2021}], | ||
| [{"Person": "Alice"}, "owns", {"Webpage": "alice.com"}, {}], [{"Person": "Bob"}, "owns", | ||
| {"Webpage": "bob.com"}, {}] NodesSchemas: ["Person", "name", {"age": "int", | ||
| "name": "text", "occupation": | ||
| "text"}], ["Webpage", "name", {"name": "text", "url": "text"}] RelationshipsSchemas :["Person", | ||
| "roommate", "Person", {"start": "int"}], ["Person", "owns", "Webpage", {}]""" |
Member
There was a problem hiding this comment.
Do we have a better way to maintain the default prompt if we make subsequent modifications and adjustments
imbajin approved these changes Jan 24, 2024
Gfreely pushed a commit to Gfreely/incubator-hugegraph-ai that referenced this pull request Jul 17, 2025
chore(hg-llm): update pandas version
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
5 participants
Add this suggestion to a batch that can be applied as a single commit. This suggestion is invalid because no changes were made to the code. Suggestions cannot be applied while the pull request is closed. Suggestions cannot be applied while viewing a subset of changes. Only one suggestion per line can be applied in a batch. Add this suggestion to a batch that can be applied as a single commit. Applying suggestions on deleted lines is not supported. You must change the existing code in this line in order to create a valid suggestion. Outdated suggestions cannot be applied. This suggestion has been applied or marked resolved. Suggestions cannot be applied from pending reviews. Suggestions cannot be applied on multi-line comments. Suggestions cannot be applied while the pull request is queued to merge. Suggestion cannot be applied right now. Please check back later.
spo triple extraction