fix: use template from Future-Outlier/flyte-custom-agent-template

Signed-off-by: mao3267 <[email protected]>

basic-custom-agent/{{cookiecutter.project_name}}/Dockerfile

@@ -17,14 +17,17 @@ RUN pip install --no-cache-dir \
 # Install Flytekit from GitHub
 RUN pip install --no-cache-dir git+https://github.com/flyteorg/flytekit.git@master
 
-# Copy and install the plugin
-COPY {{cookiecutter.project_name}} /flytekit-{{cookiecutter.project_name}}
-RUN pip install --no-cache-dir /flytekit-{{cookiecutter.project_name}}
+# Copy and install the bigquery plugin
+COPY flytekit-bigquery /flytekit-bigquery
+RUN pip install --no-cache-dir /flytekit-bigquery
+
+COPY flytekit-openai /flytekit-openai
+RUN pip install --no-cache-dir /flytekit-openai
 
 # Cleanup
 RUN apt-get purge -y build-essential git \
     && apt-get autoremove -y \
     && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
 
 # Set the default command
-CMD ["pyflyte", "serve", "agent", "--port", "8000"]
+CMD ["pyflyte", "serve", "agent", "--port", "8000"]

basic-custom-agent/{{cookiecutter.project_name}}/README.md

@@ -22,7 +22,7 @@ docker buildx build --platform linux/amd64 -t localhost:30000/flyteagent:custom-
 docker run -it localhost:30000/flyteagent:custom-agent
 ```
 
-4. Check the logs (sensor is created by flytekit, bigquery is created by the custom agent)
+4. Check the logs (sensor is created by flytekit, bigquery and openai is created by the custom agent)
 ```
 (dev) future@outlier ~ % docker run -it localhost:30000/flyteagent:custom-agent
     
@@ -34,6 +34,7 @@ Starting up the server to expose the prometheus metrics...
 ┃ Agent Name     ┃ Support Task Types            ┃ Is Sync ┃
 ┡━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━┩
 │ Sensor         │ sensor (v0)                   │ False   │
+│ ChatGPT Agent  │ chatgpt (v0)                  │ True    │
 │ Bigquery Agent │ bigquery_query_job_task (v0)  │ False   │
 └────────────────┴───────────────────────────────┴─────────┘
-```
+```

basic-custom-agent/{{cookiecutter.project_name}}/flytekit-bigquery/README.md

@@ -0,0 +1,11 @@
+# Flytekit BigQuery Plugin
+
+BigQuery enables us to build data-intensive applications without operational burden. Flyte backend can be connected with the BigQuery service. Once enabled, it can allow you to query a BigQuery table.
+
+To install the plugin, run the following command:
+
+```bash
+pip install flytekitplugins-bigquery
+```
+
+To configure BigQuery in the Flyte deployment's backend, follow the [configuration guide](https://docs.flyte.org/en/latest/deployment/plugin_setup/gcp/bigquery.html#deployment-plugin-setup-gcp-bigquery).

basic-custom-agent/{{cookiecutter.project_name}}/flytekitplugins/{{cookiecutter.project_name}}/__init__.py → basic-custom-agent/{{cookiecutter.project_name}}/flytekit-bigquery/flytekitplugins/bigquery/__init__.py

@@ -13,4 +13,4 @@
 """
 
 from .agent import BigQueryAgent
-from .task import BigQueryConfig, BigQueryTask
+from .task import BigQueryConfig, BigQueryTask

basic-custom-agent/{{cookiecutter.project_name}}/flytekitplugins/{{cookiecutter.project_name}}/agent.py → basic-custom-agent/{{cookiecutter.project_name}}/flytekit-bigquery/flytekitplugins/bigquery/agent.py

@@ -94,4 +94,4 @@ def delete(self, resource_meta: BigQueryMetadata, **kwargs):
         client.cancel_job(resource_meta.job_id, resource_meta.project, resource_meta.location)
 
 
-AgentRegistry.register(BigQueryAgent())
+AgentRegistry.register(BigQueryAgent())

basic-custom-agent/{{cookiecutter.project_name}}/flytekitplugins/{{cookiecutter.project_name}}/task.py → basic-custom-agent/{{cookiecutter.project_name}}/flytekit-bigquery/flytekitplugins/bigquery/task.py

@@ -82,4 +82,4 @@ def get_custom(self, settings: SerializationSettings) -> Dict[str, Any]:
 
     def get_sql(self, settings: SerializationSettings) -> Optional[_task_model.Sql]:
         sql = _task_model.Sql(statement=self.query_template, dialect=_task_model.Sql.Dialect.ANSI)
-        return sql
+        return sql

basic-custom-agent/{{cookiecutter.project_name}}/setup.py → basic-custom-agent/{{cookiecutter.project_name}}/flytekit-bigquery/setup.py

@@ -37,4 +37,4 @@
         "Topic :: Software Development :: Libraries :: Python Modules",
     ],
     entry_points={"flytekit.plugins": [f"{PLUGIN_NAME}=flytekitplugins.{PLUGIN_NAME}"]},
-)
+)