diff --git a/.env.template b/.env.template index 6be5f518b..085dd54cf 100644 --- a/.env.template +++ b/.env.template @@ -1,20 +1,25 @@ -OPENAI_API_KEY=sk -WEAVIATE_URL = -WEAVIATE_API_KEY = -ENV = docker -POSTGRES_USER = bla -POSTGRES_PASSWORD = bla -POSTGRES_DB = bubu -POSTGRES_HOST = localhost -POSTGRES_HOST_DOCKER = postgres -COG_ARCH_DIR = cognee -GRAPH_DB_URL = -GRAPH_DB_PW = -GRAPH_DB_USER = -AWS_ACCESS_KEY_ID = -AWS_SECRET_ACCESS_KEY = -QDRANT_API_KEY = -QDRANT_API_URL = -DB_TYPE = sqlite -GRAPHISTRY_USERNAME = xxx -GRAPHISTRY_PASSWORD = xxx +ENV="local" +TOKENIZERS_PARALLELISM="false" +LLM_API_KEY= + +GRAPHISTRY_USERNAME= +GRAPHISTRY_PASSWORD= + +SENTRY_REPORTING_URL= + +GRAPH_DATABASE_PROVIDER="neo4j" # or "networkx" +# Not needed if using networkx +GRAPH_DATABASE_URL= +GRAPH_DATABASE_USERNAME= +GRAPH_DATABASE_PASSWORD= + +VECTOR_ENGINE_PROVIDER="qdrant" # or "weaviate" or "lancedb" +# Not needed if using "lancedb" +VECTOR_DB_URL= +VECTOR_DB_KEY= + +DB_HOST=127.0.0.1 +DB_PORT=5432 +DB_USERNAME=cognee +DB_PASSWORD=cognee +DB_DATABASE=cognee_db diff --git a/.github/workflows/test_neo4j.yml b/.github/workflows/test_neo4j.yml index fe1bc1364..ae06fe458 100644 --- a/.github/workflows/test_neo4j.yml +++ b/.github/workflows/test_neo4j.yml @@ -80,19 +80,14 @@ jobs: - name: Run default Neo4j env: - OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }} + ENV: 'dev' LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} GRAPH_DATABASE_URL: ${{ secrets.NEO4J_API_URL }} GRAPH_DATABASE_PASSWORD: ${{ secrets.NEO4J_API_KEY }} GRAPH_DATABASE_USERNAME: "neo4j" - ENV: 'dev' - POSTGRES_USER: cognee - POSTGRES_PASSWORD: cognee - POSTGRES_DB: cognee_db - POSTGRES_HOST: localhost - POSTGRES_PORT: 5432 - DESTINATION__POSTGRES__CREDENTIALS__HOST: 127.0.0.1 - DESTINATION__POSTGRES__CREDENTIALS__USERNAME: cognee - DESTINATION__POSTGRES__CREDENTIALS__PASSWORD: cognee - DESTINATION__POSTGRES__CREDENTIALS__DATABASE: cognee_db + DB_USER: cognee + DB_PASSWORD: cognee + DB_NAME: cognee_db + DB_HOST: localhost + DB_PORT: 5432 run: poetry run python ./cognee/tests/test_neo4j.py diff --git a/.github/workflows/test_python_3_10.yml b/.github/workflows/test_python_3_10.yml index 0a2452a6d..88b4af1ab 100644 --- a/.github/workflows/test_python_3_10.yml +++ b/.github/workflows/test_python_3_10.yml @@ -93,15 +93,13 @@ jobs: - name: Run default basic pipeline env: - OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }} - LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} ENV: 'dev' - PGUSER: cognee - DESTINATION__POSTGRES__CREDENTIALS__HOST: localhost - DESTINATION__POSTGRES__CREDENTIALS__USERNAME: cognee - DESTINATION__POSTGRES__CREDENTIALS__PASSWORD: cognee - DESTINATION__POSTGRES__CREDENTIALS__DATABASE: cognee_db - DESTINATION__POSTGRES__CREDENTIALS__PORT: 5432 + LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} + DB_HOST: localhost + DB_USERNAME: cognee + DB_PASSWORD: cognee + DB_DATABASE: cognee_db + DB_PORT: 5432 run: poetry run python ./cognee/tests/test_library.py - name: Clean up disk space diff --git a/.github/workflows/test_python_3_11.yml b/.github/workflows/test_python_3_11.yml index adc9a5db4..eb1e07f4d 100644 --- a/.github/workflows/test_python_3_11.yml +++ b/.github/workflows/test_python_3_11.yml @@ -93,15 +93,13 @@ jobs: - name: Run default basic pipeline env: - OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }} - LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} ENV: 'dev' - PGUSER: cognee - DESTINATION__POSTGRES__CREDENTIALS__HOST: localhost - DESTINATION__POSTGRES__CREDENTIALS__USERNAME: cognee - DESTINATION__POSTGRES__CREDENTIALS__PASSWORD: cognee - DESTINATION__POSTGRES__CREDENTIALS__DATABASE: cognee_db - DESTINATION__POSTGRES__CREDENTIALS__PORT: 5432 + LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} + DB_HOST: localhost + DB_USERNAME: cognee + DB_PASSWORD: cognee + DB_DATABASE: cognee_db + DB_PORT: 5432 run: poetry run python ./cognee/tests/test_library.py - name: Clean up disk space diff --git a/.github/workflows/test_python_3_9.yml b/.github/workflows/test_python_3_9.yml index 6c679ce48..7abe8cfa3 100644 --- a/.github/workflows/test_python_3_9.yml +++ b/.github/workflows/test_python_3_9.yml @@ -93,15 +93,13 @@ jobs: - name: Run default basic pipeline env: - OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }} - LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} ENV: 'dev' - PGUSER: cognee - DESTINATION__POSTGRES__CREDENTIALS__HOST: localhost - DESTINATION__POSTGRES__CREDENTIALS__USERNAME: cognee - DESTINATION__POSTGRES__CREDENTIALS__PASSWORD: cognee - DESTINATION__POSTGRES__CREDENTIALS__DATABASE: cognee_db - DESTINATION__POSTGRES__CREDENTIALS__PORT: 5432 + LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} + DB_HOST: localhost + DB_USERNAME: cognee + DB_PASSWORD: cognee + DB_DATABASE: cognee_db + DB_PORT: 5432 run: poetry run python ./cognee/tests/test_library.py - name: Clean up disk space diff --git a/.github/workflows/test_qdrant.yml b/.github/workflows/test_qdrant.yml index 105da7d88..627807b61 100644 --- a/.github/workflows/test_qdrant.yml +++ b/.github/workflows/test_qdrant.yml @@ -80,18 +80,13 @@ jobs: - name: Run default Qdrant env: - OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }} + ENV: 'dev' LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} VECTOR_DB_URL: ${{ secrets.QDRANT_API_URL }} VECTOR_DB_KEY: ${{ secrets.QDRANT_API_KEY }} - ENV: 'dev' - POSTGRES_USER: cognee - POSTGRES_PASSWORD: cognee - POSTGRES_DB: cognee_db - POSTGRES_HOST: localhost - POSTGRES_PORT: 5432 - DESTINATION__POSTGRES__CREDENTIALS__HOST: 127.0.0.1 - DESTINATION__POSTGRES__CREDENTIALS__USERNAME: cognee - DESTINATION__POSTGRES__CREDENTIALS__PASSWORD: cognee - DESTINATION__POSTGRES__CREDENTIALS__DATABASE: cognee_db + DB_USER: cognee + DB_PASSWORD: cognee + DB_NAME: cognee_db + DB_HOST: localhost + DB_PORT: 5432 run: poetry run python ./cognee/tests/test_qdrant.py diff --git a/.github/workflows/test_weaviate.yml b/.github/workflows/test_weaviate.yml index 39bde5514..6d8849424 100644 --- a/.github/workflows/test_weaviate.yml +++ b/.github/workflows/test_weaviate.yml @@ -80,18 +80,13 @@ jobs: - name: Run default Weaviate env: - OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }} + ENV: 'dev' LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} VECTOR_DB_URL: ${{ secrets.WEAVIATE_API_URL }} VECTOR_DB_KEY: ${{ secrets.WEAVIATE_API_KEY }} - ENV: 'dev' - POSTGRES_USER: cognee - POSTGRES_PASSWORD: cognee - POSTGRES_DB: cognee_db - POSTGRES_HOST: localhost - POSTGRES_PORT: 5432 - DESTINATION__POSTGRES__CREDENTIALS__HOST: 127.0.0.1 - DESTINATION__POSTGRES__CREDENTIALS__USERNAME: cognee - DESTINATION__POSTGRES__CREDENTIALS__PASSWORD: cognee - DESTINATION__POSTGRES__CREDENTIALS__DATABASE: cognee_db + DB_USER: cognee + DB_PASSWORD: cognee + DB_NAME: cognee_db + DB_HOST: localhost + DB_PORT: 5432 run: poetry run python ./cognee/tests/test_weaviate.py diff --git a/cognee/base_config.py b/cognee/base_config.py index 9d3f4089d..d2245ef6e 100644 --- a/cognee/base_config.py +++ b/cognee/base_config.py @@ -9,8 +9,6 @@ class BaseConfig(BaseSettings): monitoring_tool: object = MonitoringTool.LANGFUSE graphistry_username: Optional[str] = None graphistry_password: Optional[str] = None - aws_access_key_id: Optional[str] = None - aws_secret_access_key: Optional[str] = None model_config = SettingsConfigDict(env_file = ".env", extra = "allow")