codellama

/

CodeLlama-70b-hf

like 312

Follow

Code Llama 514

Input validation error: inputs tokens + max_new_tokens must be <= 1512. Given: 9959 inputs tokens and 512 max_new_tokens

Code

llm = HuggingFaceEndpoint(
    endpoint_url=f"{your_endpoint_url}",
    max_new_tokens=512,
    top_k=10,
    top_p=0.1,
    typical_p=0.95,
    temperature=0.01,
    repetition_penalty=1.03,
)
db_chain = create_sql_query_chain(llm=llm, db=db)
user_query = "<BLAH BLAH>"
context = """
"""
prompt = f"""Please that your job is to write an SQL query to extract this data from a Postgres database and not to actually create visualizations.The visualization creation will be done later.Given an input question, first create a syntactically correct PostgreSQL query to run. DO NOT include any extra content.
Info About Dataset {context}
            Use the following format:
            Question: "Question here"
            SQL Query to run
            Question: {user_query}"""
def find_sql_queries(text):
    # Regex pattern to match basic SQL queries
    words_to_remove = ["SQLQuery:", "sql"]
    for word_to_remove in words_to_remove:
        text = text.replace(word_to_remove, "")
    text = text[text.find("SELECT"): text.find(";")]
    return text

result = find_sql_queries(db_chain.invoke({"question": prompt}))
print(result)

I deployed codellama 70B model