Fix

microsoft · Apr 1, 2024 · 42e3a21 · 42e3a21
1 parent fd5e643
commit 42e3a21
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/promptflow-core/promptflow/_utils/user_agent_utils.py b/src/promptflow-core/promptflow/_utils/user_agent_utils.py
@@ -61,7 +61,7 @@ def setup_user_agent_to_operation_context(user_agent):
 
 def append_promptflow_package_ua(operation_context: OperationContext):
     try:
-        from promptflow.core._version import VERSION as PF_VERSION
+        from promptflow._version import VERSION as PF_VERSION
 
         operation_context.append_user_agent(f"promptflow/{PF_VERSION}")
     except ImportError:

diff --git a/src/promptflow/tests/executor/unittests/_utils/test_exception_utils.py b/src/promptflow/tests/executor/unittests/_utils/test_exception_utils.py
@@ -14,7 +14,7 @@
     last_frame_info,
     remove_suffix,
 )
-from promptflow.core._version import VERSION
+from promptflow._version import VERSION
 from promptflow.exceptions import (
     ErrorTarget,
     PromptflowException,

diff --git a/src/promptflow/tests/executor/unittests/executor/_service/utils/test_service_utils.py b/src/promptflow/tests/executor/unittests/executor/_service/utils/test_service_utils.py
@@ -7,7 +7,7 @@
 
 from promptflow._utils.exception_utils import ExceptionPresenter, JsonSerializedPromptflowException, ResponseCode
 from promptflow._utils.logger_utils import bulk_logger, flow_logger, logger, service_logger
-from promptflow.core._version import VERSION as PF_VERSION
+from promptflow._version import VERSION as PF_VERSION
 from promptflow.core._version import __version__ as PF_CORE_VERSION
 from promptflow.executor._service._errors import ExecutionTimeoutError
 from promptflow.executor._service.contracts.execution_request import BaseExecutionRequest, FlowExecutionRequest