86 lines
3 KiB
Python
86 lines
3 KiB
Python
DEFAULT_TOOLS = [
|
|
{
|
|
"type": "function",
|
|
"name": "search",
|
|
"description": "Search for information within the knowledge graph",
|
|
"parameters": {
|
|
"type": "object",
|
|
"properties": {
|
|
"search_query": {
|
|
"type": "string",
|
|
"description": "The query to search for in the knowledge graph",
|
|
},
|
|
"search_type": {
|
|
"type": "string",
|
|
"description": "Type of search to perform",
|
|
"enum": [
|
|
"INSIGHTS",
|
|
"CODE",
|
|
"GRAPH_COMPLETION",
|
|
"SEMANTIC",
|
|
"NATURAL_LANGUAGE",
|
|
],
|
|
},
|
|
"top_k": {
|
|
"type": "integer",
|
|
"description": "Maximum number of results to return",
|
|
"default": 10,
|
|
},
|
|
"datasets": {
|
|
"type": "array",
|
|
"items": {"type": "string"},
|
|
"description": "Optional list of dataset names to search within",
|
|
},
|
|
},
|
|
"required": ["search_query"],
|
|
},
|
|
},
|
|
{
|
|
"type": "function",
|
|
"name": "cognify_text",
|
|
"description": "Convert text into a knowledge graph or process all added content",
|
|
"parameters": {
|
|
"type": "object",
|
|
"properties": {
|
|
"text": {
|
|
"type": "string",
|
|
"description": "Text content to be converted into a knowledge graph",
|
|
},
|
|
"graph_model_name": {
|
|
"type": "string",
|
|
"description": "Name of the graph model to use",
|
|
},
|
|
"graph_model_file": {
|
|
"type": "string",
|
|
"description": "Path to a custom graph model file",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
"type": "function",
|
|
"name": "prune",
|
|
"description": "Remove unnecessary or outdated information from the knowledge graph",
|
|
"parameters": {
|
|
"type": "object",
|
|
"properties": {
|
|
"prune_strategy": {
|
|
"type": "string",
|
|
"enum": ["light", "moderate", "aggressive"],
|
|
"description": "Strategy for pruning the knowledge graph",
|
|
"default": "moderate",
|
|
},
|
|
"min_confidence": {
|
|
"type": "number",
|
|
"description": "Minimum confidence score to retain (0-1)",
|
|
"minimum": 0,
|
|
"maximum": 1,
|
|
},
|
|
"older_than": {
|
|
"type": "string",
|
|
"description": "ISO date string - prune nodes older than this date",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
]
|