Grand Overhaul: Refactored Core Structure, Introduced New Features, E…

.env.example

@@ -0,0 +1 @@
+GRAPHRAG_QUERY__CHAT_LLM__API_KEY=Your API Key

graphrag.example.yml

@@ -0,0 +1,53 @@
+chat_llm:
+  model: Chat LLM Model
+  api_key: Your API Key
+  base_url: null
+  organization: null
+  timeout: null
+  max_retries: null
+  kwargs: null
+
+embedding:
+  model: Text Embedding Model
+  api_key: Your API Key
+  base_url: null
+  timeout: null
+  max_retries: null
+  max_tokens: null
+  token_encoder: null
+  kwargs: null
+
+logging:
+  enabled: true
+  level: INFO
+  format: null
+  out_file: null
+  err_file: null
+  rotation: null
+  retention: null
+  serialize: null
+  kwargs: null
+
+context:
+  directory: Your Context Directory
+  kwargs: null
+
+local_search:
+  sys_prompt: null
+  community_level: null
+  store_coll_name: null
+  store_uri: null
+  encoding_model: null
+  kwargs: null
+
+global_search:
+  map_sys_prompt: null
+  reduce_sys_prompt: null
+  community_level: null
+  allow_general_knowledge: null
+  general_knowledge_sys_prompt: null
+  no_data_answer: null
+  json_mode: null
+  max_data_tokens: null
+  encoding_model: null
+  kwargs: null

graphrag/api/__init__.py

@@ -9,21 +9,11 @@
 
 from .index_api import build_index
 from .prompt_tune_api import DocSelectionType, generate_indexing_prompts
-from .query_api import (
-    global_search,
-    global_search_streaming,
-    local_search,
-    local_search_streaming,
-)
+
 
 __all__ = [  # noqa: RUF022
     # index API
     "build_index",
-    # query API
-    "global_search",
-    "global_search_streaming",
-    "local_search",
-    "local_search_streaming",
     # prompt tuning API
     "DocSelectionType",
     "generate_indexing_prompts",