mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-13 21:21:08 +00:00
feat(build): Support docker install
This commit is contained in:
@@ -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
|
||||
|
@@ -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
|
||||
|
||||
|
||||
|
@@ -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
|
||||
|
Reference in New Issue
Block a user