docs/tutorials /sql_qa.ipynb

[zerd1y]

Checked other resources

• I added a very descriptive title to this question.
• I searched the LangChain documentation with the integrated search.
• I used the GitHub search to find a similar question and didn't find it.

Commit to Help

• I commit to help with one of those options 👆

Example Code

import os
from langchain_groq import ChatGroq
from langchain import hub
from typing_extensions import Annotated
from typing_extensions import TypedDict


os.environ["LANGCHAIN_TRACING_V2"] = "true"
os.environ["LANGCHAIN_API_KEY"] = "my_api_key"


class State(TypedDict):
    question: str
    query: str
    result: str
    answer: str


api_key = "my_api_key"
llm = ChatGroq(api_key = api_key, model="llama3-8b-8192")

from langchain_community.utilities import SQLDatabase

db = SQLDatabase.from_uri("sqlite:///C:/sqlite/Chinook.db")

query_prompt_template = hub.pull("langchain-ai/sql-query-system-prompt")

assert len(query_prompt_template.messages) == 1
query_prompt_template.messages[0].pretty_print()


class QueryOutput(TypedDict):
    """Generated SQL query."""

    query: Annotated[str, ..., "Syntactically valid SQL query."]


def write_query(state: State):
    """Generate SQL query to fetch information."""
    prompt = query_prompt_template.invoke(
        {
            "dialect": db.dialect,
            "top_k": 10,
            "table_info": db.get_table_info(),
            "input": state["question"],
        }
    )
    structured_llm = llm.with_structured_output(QueryOutput)
    result = structured_llm.invoke(prompt)
    return {"query": result["query"]}

write_query({"question": "How many Employees are there?"})

Description

I want him to output the results in the official documentation(just like this)，what should i do next？

When I run it directly, it looks like this：
(Anaconda Prompt)

(VS code terminal)

What should I do next?

System Info

langchain==0.3.8
langchain-chroma==0.1.4
langchain-community==0.3.8
langchain-core==0.3.21
langchain-groq==0.2.1
langchain-huggingface==0.1.2
langchain-openai==0.2.8
langchain-text-splitters==0.3.2
langchainhub==0.1.21

windows 11

python 3.12.7

[feijoes]

@zerd1y What write_query returns?

print(write_query({"question": "How many Employees are there?"})) # ?