cwyodmodules 0.3.47__py3-none-any.whl → 0.3.48__py3-none-any.whl

cwyodmodules/batch/utilities/helpers/env_helper.py

@@ -7,7 +7,7 @@ from ..orchestrator.orchestration_strategy import OrchestrationStrategy
 from ..helpers.config.conversation_flow import ConversationFlow
 from ..helpers.config.database_type import DatabaseType
 
-from mgmt_config import logger, identity, keyvault, create_keyvault_client
+from mgmt_config import logger, identity, keyvault, head_keyvault
 
 
 
@@ -29,23 +29,23 @@ class EnvHelper:
 
         logger.info("Initializing EnvHelper")
         
+        if not keyvault:
+            raise ValueError("keyvault is not configured. Please set 'key_vault_uri' environment variable.")
+        
+        if not head_keyvault:
+            raise ValueError("head_keyvault is not configured. Please set 'head_key_vault_uri' environment variable.")
+
         # Wrapper for Azure Key Vault
         os.environ["APPLICATIONINSIGHTS_ENABLED"] = "true"
 
-        self.secretHelper = keyvault
-
-        self.secretHelperHead = create_keyvault_client(
-            vault_url=os.environ["head_key_vault_uri"]
-        )
-
-        # self.secretHelper = SecretHelper(
+        # keyvault = SecretHelper(
         #     keyvault_uri="https://www.kv-main-cwyod-res1.vault.azure.net/"
         # )
-        # self.secretHelperHead = SecretHelper(
+        # head_keyvault = SecretHelper(
         #     keyvault_uri="https://www.kv-main-cwyod-hd-res1.vault.azure.net/"
         # )
 
-        self.LOGLEVEL = self.secretHelper.get_secret("logging-level")
+        self.LOGLEVEL = keyvault.get_secret("logging-level")
         self.LOG_EXECUTION = self.get_env_var_bool(
                 "LOG_EXECUTION", "True"
             )
@@ -57,18 +57,18 @@ class EnvHelper:
             )
 
         # Azure
-        self.AZURE_SUBSCRIPTION_ID = self.secretHelper.get_secret("subscription-id")
-        self.AZURE_RESOURCE_GROUP = self.secretHelper.get_secret("resource-group-name")
-        self.AZURE_HEAD_RESOURCE_GROUP = self.secretHelperHead.get_secret(
+        self.AZURE_SUBSCRIPTION_ID = keyvault.get_secret("subscription-id")
+        self.AZURE_RESOURCE_GROUP = keyvault.get_secret("resource-group-name")
+        self.AZURE_HEAD_RESOURCE_GROUP = head_keyvault.get_secret(
             "resource-group-name"
         )
-        self.AZURE_RESOURCE_ENVIRONMENT = self.secretHelper.get_secret(
+        self.AZURE_RESOURCE_ENVIRONMENT = keyvault.get_secret(
             "resource-group-environment"
         )
         self.AZURE_RESOURCE_PRIVATE = (
-            self.secretHelper.get_secret("run-private-endpoint").lower() == "true"
+            keyvault.get_secret("run-private-endpoint").lower() == "true"
         )
-        self.PROJECT_CODE = self.secretHelper.get_secret("project-code")
+        self.PROJECT_CODE = keyvault.get_secret("project-code")
         self.APP_NAME = os.getenv("REFLECTION_NAME", "Default")
         self.POSTGRESQL_NAME = (
             f"psql-main-{self.PROJECT_CODE}-{self.AZURE_RESOURCE_ENVIRONMENT}"
@@ -83,7 +83,7 @@ class EnvHelper:
             f"https://{self.AZURE_BLOB_ACCOUNT_NAME}.blob.core.windows.net/"
         )
 
-        self.AZURE_FUNCTION_APP_ENDPOINT = self.secretHelper.get_secret(
+        self.AZURE_FUNCTION_APP_ENDPOINT = keyvault.get_secret(
             f"func-backend-{self.PROJECT_CODE}-{self.AZURE_RESOURCE_ENVIRONMENT}-endpoint"
         )
         self.AZURE_BLOB_CONTAINER_NAME = "documents"
@@ -115,7 +115,7 @@ class EnvHelper:
                 self.AZURE_COSMOSDB_CONVERSATIONS_CONTAINER = os.getenv(
                     "AZURE_COSMOSDB_CONVERSATIONS_CONTAINER_NAME", ""
                 )
-            self.AZURE_COSMOSDB_ACCOUNT_KEY = self.secretHelper.get_secret(
+            self.AZURE_COSMOSDB_ACCOUNT_KEY = keyvault.get_secret(
                 "AZURE_COSMOSDB_ACCOUNT_KEY"
             )
             self.AZURE_COSMOSDB_ENABLE_FEEDBACK = (
@@ -142,10 +142,10 @@ class EnvHelper:
                 self.POSTGRESQL_HOST = azure_postgresql_info.get("host", "")
             else:
                 self.POSTGRESQL_USER = "cwyod_project_uai"
-                self.POSTGRESQL_DATABASE = self.secretHelper.get_secret(
+                self.POSTGRESQL_DATABASE = keyvault.get_secret(
                     f"{self.POSTGRESQL_NAME}-default-database-name"
                 )
-                self.POSTGRESQL_HOST = self.secretHelper.get_secret(
+                self.POSTGRESQL_HOST = keyvault.get_secret(
                     f"{self.POSTGRESQL_NAME}-server-name"
                 )
             # Ensure integrated vectorization is disabled for PostgreSQL
@@ -177,11 +177,11 @@ class EnvHelper:
             True if self.AZURE_OPENAI_STREAM.lower() == "true" else False
         )
 
-        # self.AZURE_COMPUTER_VISION_NAME = self.secretHelperHead.get_secret(
+        # self.AZURE_COMPUTER_VISION_NAME = head_keyvault.get_secret(
         #     "cognitive-kind-ComputerVision"
         # )
         self.AZURE_COMPUTER_VISION_NAME = ""
-        # self.AZURE_COMPUTER_VISION_ENDPOINT = self.secretHelperHead.get_secret(
+        # self.AZURE_COMPUTER_VISION_ENDPOINT = head_keyvault.get_secret(
         #     f"{self.AZURE_COMPUTER_VISION_NAME}-endpoint"
         # )
         self.AZURE_COMPUTER_VISION_ENDPOINT = ""
@@ -197,16 +197,16 @@ class EnvHelper:
             self.AZURE_OPENAI_API_KEY = ""
             self.AZURE_COMPUTER_VISION_KEY = None
         else:
-            self.AZURE_SEARCH_KEY = self.secretHelper.get_secret("AZURE_SEARCH_KEY")
-            self.AZURE_OPENAI_API_KEY = self.secretHelper.get_secret(
+            self.AZURE_SEARCH_KEY = keyvault.get_secret("AZURE_SEARCH_KEY")
+            self.AZURE_OPENAI_API_KEY = keyvault.get_secret(
                 "AZURE_OPENAI_API_KEY"
             )
-            self.AZURE_COMPUTER_VISION_KEY = self.secretHelper.get_secret(
+            self.AZURE_COMPUTER_VISION_KEY = keyvault.get_secret(
                 "AZURE_COMPUTER_VISION_KEY"
             )
 
         # Set env for Azure OpenAI
-        self.AZURE_AI_SERVICES_NAME = self.secretHelperHead.get_secret(
+        self.AZURE_AI_SERVICES_NAME = head_keyvault.get_secret(
             "cognitive-kind-AIServices"
         )
         self.AZURE_OPENAI_ENDPOINT = (
@@ -226,10 +226,10 @@ class EnvHelper:
 
         # Azure Form Recognizer
 
-        self.AZURE_FORM_RECOGNIZER_NAME = self.secretHelper.get_secret(
+        self.AZURE_FORM_RECOGNIZER_NAME = keyvault.get_secret(
             "cognitive-kind-FormRecognizer"
         )
-        self.AZURE_FORM_RECOGNIZER_ENDPOINT = self.secretHelper.get_secret(
+        self.AZURE_FORM_RECOGNIZER_ENDPOINT = keyvault.get_secret(
             f"{self.AZURE_FORM_RECOGNIZER_NAME}-endpoint"
         )
 
@@ -238,18 +238,18 @@ class EnvHelper:
         self.APPLICATIONINSIGHTS_ENABLED = "True"
 
         # Azure AI Content Safety
-        self.AZURE_CONTENT_SAFETY_NAME = self.secretHelperHead.get_secret(
+        self.AZURE_CONTENT_SAFETY_NAME = head_keyvault.get_secret(
             "cognitive-kind-ContentSafety"
         )
-        self.AZURE_CONTENT_SAFETY_ENDPOINT = self.secretHelperHead.get_secret(
+        self.AZURE_CONTENT_SAFETY_ENDPOINT = head_keyvault.get_secret(
             f"{self.AZURE_CONTENT_SAFETY_NAME}-endpoint"
         )
 
         # Speech Service
-        self.AZURE_SPEECH_SERVICE_NAME = self.secretHelperHead.get_secret(
+        self.AZURE_SPEECH_SERVICE_NAME = head_keyvault.get_secret(
             "cognitive-kind-SpeechServices"
         )
-        self.AZURE_SPEECH_ENDPOINT = self.secretHelperHead.get_secret(
+        self.AZURE_SPEECH_ENDPOINT = head_keyvault.get_secret(
             f"{self.AZURE_SPEECH_SERVICE_NAME}-endpoint"
         )
 
@@ -262,7 +262,7 @@ class EnvHelper:
         self.AZURE_SPEECH_REGION_ENDPOINT = (
             f"https://{self.AZURE_SPEECH_SERVICE_REGION}.api.cognitive.microsoft.com/"
         )
-        self.AZURE_SPEECH_KEY = self.secretHelperHead.get_secret(f"{self.AZURE_SPEECH_SERVICE_NAME}-key") # changed from None to allow private endpoint communication
+        self.AZURE_SPEECH_KEY = head_keyvault.get_secret(f"{self.AZURE_SPEECH_SERVICE_NAME}-key") # changed from None to allow private endpoint communication
 
         self.LOAD_CONFIG_FROM_BLOB_STORAGE = self.get_env_var_bool(
             "LOAD_CONFIG_FROM_BLOB_STORAGE"

{cwyodmodules-0.3.47.dist-info → cwyodmodules-0.3.48.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cwyodmodules
-Version: 0.3.47
+Version: 0.3.48
 Summary: Add your description here
 Author-email: Patrik <patrikhartl@gmail.com>
 Classifier: Operating System :: OS Independent
@@ -40,7 +40,7 @@ Requires-Dist: azure-search-documents==11.6.0b4
 Requires-Dist: semantic-kernel==1.3.0
 Requires-Dist: pydantic==2.7.4
 Requires-Dist: pandas>=2.2.3
-Requires-Dist: azpaddypy>=0.3.5
+Requires-Dist: azpaddypy>=0.3.7
 Dynamic: license-file
 
 # paddypy

{cwyodmodules-0.3.47.dist-info → cwyodmodules-0.3.48.dist-info}/RECORD

@@ -37,7 +37,7 @@ cwyodmodules/batch/utilities/helpers/azure_postgres_helper_light_rag.py,sha256=M
 cwyodmodules/batch/utilities/helpers/azure_search_helper.py,sha256=vIIMEck1wPg9oRlWweE2gSZ1nUYc_tmEe4QeFlsrwKk,11314
 cwyodmodules/batch/utilities/helpers/document_chunking_helper.py,sha256=2MZOjW-fHXgYijP3m9O-nizOlk96Yg0axyxT0K6fTnM,725
 cwyodmodules/batch/utilities/helpers/document_loading_helper.py,sha256=2HBEl3vW-_PKbX5pPntTC_R5eToTk2Qb-q3M4Mt6hCU,603
-cwyodmodules/batch/utilities/helpers/env_helper.py,sha256=avZCU8eZOjrhGLu8LcjQr_kjoLzkL_75UemCGQTINxY,15827
+cwyodmodules/batch/utilities/helpers/env_helper.py,sha256=0wJTaoMAR8rBo1-q2TQx2xqkYktbgk9EsUmZSsvwIJ0,15727
 cwyodmodules/batch/utilities/helpers/lightrag_helper.py,sha256=7lb9JMm5IohsO73LWo5bWmlzWCGYNsx_fYl-aFdwATQ,3845
 cwyodmodules/batch/utilities/helpers/llm_helper.py,sha256=lHLYrUidtaemmKrVbWoo7oIvwluUoPUk16U5lV-YIX8,8282
 cwyodmodules/batch/utilities/helpers/orchestrator_helper.py,sha256=mCcZyMFG0otnw9gzWd-PYocHmDdFDVg-RT9oDPiDZPk,897
@@ -109,8 +109,8 @@ cwyodmodules/graphrag/query/generate.py,sha256=BZiB6iw7PkIovw-CyYFogMHnDxK0Qu_4u
 cwyodmodules/graphrag/query/graph_search.py,sha256=95h3ecSWx4864XgKABtG0fh3Nk8HkqJVzoCrO8daJ-Y,7724
 cwyodmodules/graphrag/query/types.py,sha256=1Iq1dp4I4a56_cuFjOZ0NTgd0A2_MpVFznp_czgt6cI,617
 cwyodmodules/graphrag/query/vector_search.py,sha256=9Gwu9LPjtoAYUU8WKqCvbCHAIg3dpk71reoYd1scLnQ,1807
-cwyodmodules-0.3.47.dist-info/licenses/LICENSE,sha256=UqBDTipijsSW2ZSOXyTZnMsXmLoEHTgNEM0tL4g-Sso,1150
-cwyodmodules-0.3.47.dist-info/METADATA,sha256=YSDKlXiAglRuJPkC06tOr1GIB9zXKgRfGgMTKl-cFWE,2002
-cwyodmodules-0.3.47.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-cwyodmodules-0.3.47.dist-info/top_level.txt,sha256=99RENLbkdRX-qpJvsxZ5AfmTL5s6shSaKOWYpz1vwzg,13
-cwyodmodules-0.3.47.dist-info/RECORD,,
+cwyodmodules-0.3.48.dist-info/licenses/LICENSE,sha256=UqBDTipijsSW2ZSOXyTZnMsXmLoEHTgNEM0tL4g-Sso,1150
+cwyodmodules-0.3.48.dist-info/METADATA,sha256=9YMDuaHhW4n_XMuMVucVnaefjDVdVch9u7s6bl4O5Bg,2002
+cwyodmodules-0.3.48.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+cwyodmodules-0.3.48.dist-info/top_level.txt,sha256=99RENLbkdRX-qpJvsxZ5AfmTL5s6shSaKOWYpz1vwzg,13
+cwyodmodules-0.3.48.dist-info/RECORD,,