-
Notifications
You must be signed in to change notification settings - Fork 5
Feat/mcp/new tools #54
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
cvauclair
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nice work! Small thing to change but apart from that it looks good to merge
| } | ||
| } | ||
|
|
||
| impl<T: FromAttributes> Query<Vec<Relation>> for FindPathQuery<T> { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
In queries that return data, the generic type T is used to indicate what shape of data will be returned. However, since here were returning a "hardcoded" Vec<Relation> as the data, we don't actually need T.
There are two possible fixes:
- (recommended) Change the
impltoimpl Query<Vec<Relation>> for FindPathQuery<Vec<Relation>> - Remove
TfromFindPathQuery(not ideal since we might want to support more return types in the future)
| skip: Option<usize>, | ||
| space_id: Option<PropFilter<String>>, | ||
| version: VersionFilter, | ||
| _phantom: std::marker::PhantomData<EntityNode>, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Since FindPathQuery does not have a generic type parameter anymore, we don't need to have a _phantom field!
cvauclair
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nice work!
Add new tools for MCP server to query the knowledge graph
Modified: get_entity_info
Added: search_relation_between_2_entities
Added: get_entity_by_attribute
new function for resolving Id of entity to their name
embeddings are now available for the sample data