♻️Refactor log level

badDevBuild · Jul 17, 2023 · 83e2be7 · 83e2be7
1 parent fbdf536
commit 83e2be7
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/platform/reworkd_platform/__main__.py b/platform/reworkd_platform/__main__.py
@@ -11,7 +11,7 @@ def main() -> None:
         host=settings.host,
         port=settings.port,
         reload=settings.reload,
-        log_level=settings.log_level.value.lower(),
+        log_level=settings.log_level.lower(),
         factory=True,
     )
 

diff --git a/platform/reworkd_platform/logging.py b/platform/reworkd_platform/logging.py
@@ -59,5 +59,5 @@ def configure_logging() -> None:  # pragma: no cover
     logger.remove()
     logger.add(
         sys.stdout,
-        level=settings.log_level.value,
+        level=settings.log_level,
     )
diff --git a/platform/reworkd_platform/services/pinecone/pinecone.py b/platform/reworkd_platform/services/pinecone/pinecone.py
@@ -3,9 +3,9 @@
 import uuid
 from typing import Any, Dict, List
 
-import pinecone
 from langchain.embeddings import OpenAIEmbeddings
 from langchain.embeddings.base import Embeddings
+from pinecone import Index  # import doesnt work on plane wifi
 from pydantic import BaseModel
 
 from reworkd_platform.settings import settings
@@ -33,7 +33,7 @@ class PineconeMemory(AgentMemory):
     """
 
     def __init__(self, index_name: str):
-        self.index = pinecone.Index(settings.pinecone_index_name)
+        self.index = Index(settings.pinecone_index_name)
         self.namespace = index_name
 
     @timed_function(level="DEBUG")