mirror of
https://github.com/zebrajr/localGPT.git
synced 2025-12-06 12:20:53 +01:00
6d73a61e5c
Removed unused import statements from various Python files to improve code clarity and reduce unnecessary dependencies.
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
from dotenv import load_dotenv
|
|
|
|
def get_agent(mode: str = "default"):
|
|
"""
|
|
Factory function to get an instance of the RAG agent based on the specified mode.
|
|
This uses local imports to prevent circular dependencies.
|
|
"""
|
|
from rag_system.agent.loop import Agent
|
|
from rag_system.utils.ollama_client import OllamaClient
|
|
from rag_system.main import PIPELINE_CONFIGS, OLLAMA_CONFIG
|
|
|
|
load_dotenv()
|
|
|
|
llm_client = OllamaClient(host=OLLAMA_CONFIG["host"])
|
|
config = PIPELINE_CONFIGS.get(mode, PIPELINE_CONFIGS['default'])
|
|
|
|
if 'storage' not in config:
|
|
config['storage'] = {
|
|
'db_path': 'lancedb',
|
|
'text_table_name': 'text_pages_default',
|
|
'image_table_name': 'image_pages'
|
|
}
|
|
|
|
agent = Agent(
|
|
pipeline_configs=config,
|
|
llm_client=llm_client,
|
|
ollama_config=OLLAMA_CONFIG
|
|
)
|
|
return agent
|
|
|
|
def get_indexing_pipeline(mode: str = "default"):
|
|
"""
|
|
Factory function to get an instance of the Indexing Pipeline.
|
|
"""
|
|
from rag_system.pipelines.indexing_pipeline import IndexingPipeline
|
|
from rag_system.main import PIPELINE_CONFIGS, OLLAMA_CONFIG
|
|
from rag_system.utils.ollama_client import OllamaClient
|
|
|
|
load_dotenv()
|
|
llm_client = OllamaClient(host=OLLAMA_CONFIG["host"])
|
|
config = PIPELINE_CONFIGS.get(mode, PIPELINE_CONFIGS['default'])
|
|
|
|
return IndexingPipeline(config, llm_client, OLLAMA_CONFIG) |