Use correct plugin in agent image (#2317)

Signed-off-by: Eduardo Apolinario <[email protected]> Co-authored-by: Eduardo Apolinario <[email protected]>
flyteorg · Apr 2, 2024 · 55f0b19 · 55f0b19
1 parent 2cbdc99
commit 55f0b19
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.agent b/Dockerfile.agent
@@ -11,7 +11,7 @@ RUN pip install prometheus-client grpcio-health-checking
 RUN pip install --no-cache-dir -U flytekit==$VERSION \
   flytekitplugins-airflow==$VERSION \
   flytekitplugins-bigquery==$VERSION \
-  flytekitplugins-chatgpt==$VERSION \
+  flytekitplugins-openai==$VERSION \
   flytekitplugins-snowflake==$VERSION \
   flytekitplugins-awssagemaker==$VERSION \
   && apt-get clean autoclean \