diff --git a/MANIFEST.in b/MANIFEST.in new file mode 100644 index 0000000..8982cfb --- /dev/null +++ b/MANIFEST.in @@ -0,0 +1 @@ +include src/agentCores/data/agent_matrix.db \ No newline at end of file diff --git a/README.md b/README.md index 3ba2de7..8e9df19 100644 --- a/README.md +++ b/README.md @@ -187,7 +187,7 @@ example usage: agent_config = { "agentCore": { "agent_id": "custom_assistant", - "models": {"large_language_model": "gpt-4"}, + "models": {"large_language_model": "phi3"}, "prompts": {"user_input_prompt": "You are a helpful assistant."} } } diff --git a/src/agentCores/agentCore1.png b/agentCore1.png similarity index 100% rename from src/agentCores/agentCore1.png rename to agentCore1.png diff --git a/src/agentCores/agentCores.py b/src/agentCores/agentCores.py index ef93bcc..af0e9d7 100644 --- a/src/agentCores/agentCores.py +++ b/src/agentCores/agentCores.py @@ -77,6 +77,7 @@ import copy from pathlib import Path from typing import Optional, Dict, Any +from pkg_resources import resource_filename from .agentMatrix import agentMatrix class agentCores: @@ -100,11 +101,16 @@ class agentCores: } def __init__(self, - db_path: str = "agent_matrix.db", + db_path: str = None, db_config: Optional[Dict] = None, template: Optional[Dict] = None): """Initialize AgentCore with optional custom configuration.""" self.current_date = time.strftime("%Y-%m-%d") + + # Use package data path if no custom path provided + if db_path is None: + db_path = resource_filename('agentCores', 'data/agent_matrix.db') + self.agent_library = agentMatrix(db_path) # Initialize template with any custom configuration diff --git a/src/agentCores/agent_matrix.db b/src/agentCores/data/agent_matrix.db similarity index 96% rename from src/agentCores/agent_matrix.db rename to src/agentCores/data/agent_matrix.db index f86c01b..f910431 100644 Binary files a/src/agentCores/agent_matrix.db and b/src/agentCores/data/agent_matrix.db differ