feat(build): Support docker install

This commit is contained in:
Fangyin Cheng
2025-03-11 15:57:52 +08:00
parent ee1ac0df8e
commit 455cdd1612
32 changed files with 1844 additions and 589 deletions

View File

@@ -232,7 +232,7 @@ with DAG("simple_nl_schema_sql_chart_example") as dag:
)
request_handle_task = RequestHandleOperator()
query_operator = MapOperator(lambda request: request["query"])
llm = OpenAILLMClient()
llm = (OpenAILLMClient(api_key=os.getenv("OPENAI_API_KEY", "your api key")),)
model_name = "gpt-3.5-turbo"
retriever_task = SchemaLinkingOperator(
connector=_create_temporary_connection(), llm=llm, model_name=model_name

View File

@@ -29,6 +29,7 @@
}'
"""
import os
from typing import Dict
from dbgpt._private.pydantic import BaseModel, Field
@@ -61,7 +62,10 @@ with DAG("dbgpt_awel_simple_rag_rewrite_example") as dag:
)
request_handle_task = RequestHandleOperator()
# build query rewrite operator
rewrite_task = QueryRewriteOperator(llm_client=OpenAILLMClient(), nums=2)
rewrite_task = QueryRewriteOperator(
llm_client=OpenAILLMClient(api_key=os.getenv("OPENAI_API_KEY", "your api key")),
nums=2,
)
trigger >> request_handle_task >> rewrite_task

View File

@@ -27,6 +27,7 @@ This example shows how to use AWEL to build a simple rag summary example.
}'
"""
import os
from typing import Dict
from dbgpt._private.pydantic import BaseModel, Field
@@ -62,7 +63,8 @@ with DAG("dbgpt_awel_simple_rag_summary_example") as dag:
knowledge_operator = KnowledgeOperator(knowledge_type=KnowledgeType.URL.name)
# build summary assembler operator
summary_operator = SummaryAssemblerOperator(
llm_client=OpenAILLMClient(), language="en"
llm_client=OpenAILLMClient(api_key=os.getenv("OPENAI_API_KEY", "your api key")),
language="en",
)
(
trigger