from onprem import LLM
import os
prompt_template = """<|start_header_id|>system<|end_header_id|>
You are a super-intelligent helpful assistant that executes instructions.<|eot_id|><|start_header_id|>user<|end_header_id|>
{prompt}<|eot_id|><|start_header_id|>assistant<|end_header_id|>
"""Use Prompts to Solve Problems
This notebook shows various examples of using OnPrem.LLM to solve different tasks.
Setup the LLM instance
In this notebook, we will use the Llama-3.1-8B model from Meta. In particular, we will use Meta-Llama-3.1-8B-Instruct-GGUF. There are different instances of this model on the Hugging Face model hub, and we will use the one from LM Studio. When selecting a model that is different than the default ones in OnPrem.LLM, it is important to inspect the model’s home page and identify the correct prompt format. The prompt format for this model is located here, and we will supply it directly to the LLM constructor along with the URL to the specific model file we want (i.e., Meta-Llama-3.1-8B-Instruct-Q4_K_M.gguf). We will offload layers to our GPU(s) to speed up inference using the n_gpu_layers parameter. (For more information on GPU acceleration, see here.) For the purposes of this notebook, we also supply temperature=0 so that there is no variability in outputs. You can increase this value for more creativity in the outputs. Note that you can change the system prompt (i.e., “You are a super-intelligent helpful assistant…”) to fit your needs.
llm = LLM(model_url='https://huggingface.co/lmstudio-community/Meta-Llama-3.1-8B-Instruct-GGUF/resolve/main/Meta-Llama-3.1-8B-Instruct-Q4_K_M.gguf',
prompt_template= prompt_template,
n_gpu_layers=-1,
temperature=0,
verbose=False)Note that, if supplying the convenience parameter, default_model='llama' to the LLM constructor, model_url and prompt_template are set automatically and do not need to be supplied as we did above.
Information Extraction
This is an example of zero-shot prompting:
prompt = """Extract the names of people in the supplied sentences. Separate the names with commas.
[Sentence]: I like Cillian Murphy's acting. Florence Pugh is great, too.
[People]:"""
saved_output = llm.prompt(prompt, stop=[])Cillian Murphy, Florence PughA more complicated example of Information Extraction using few-shot prompting:
prompt = """ Extract the Name, Current Position, and Current Company from each piece of Text.
Text: Alan F. Estevez serves as the Under Secretary of Commerce for Industry and Security. As Under Secretary, Mr. Estevez leads
the Bureau of Industry and Security, which advances U.S. national security, foreign policy, and economic objectives by ensuring an
effective export control and treaty compliance system and promoting U.S. strategic technology leadership.
A: Name: Alan F. Estevez | Current Position: Under Secretary | Current Company: Bureau of Industry and Security
Text: Pichai Sundararajan (born June 10, 1972[3][4][5]), better known as Sundar Pichai (/ˈsʊndɑːr pɪˈtʃaɪ/), is an Indian-born American
business executive.[6][7] He is the chief executive officer (CEO) of Alphabet Inc. and its subsidiary Google.[8]
A: Name: Sundar Pichai | Current Position: CEO | Current Company: Google
Now, provide the answer (A) from this Text:
Text: Norton Allan Schwartz (born December 14, 1951)[1] is a retired United States Air Force general[2] who served as the 19th Chief of Staff of the
Air Force from August 12, 2008, until his retirement in 2012.[3] He previously served as commander, United States Transportation Command from
September 2005 to August 2008. He is currently the president of the Institute for Defense Analyses, serving since January 2, 2020.[4]
A:"""
saved_output = llm.prompt(prompt, stop=[])Name: Norton Allan Schwartz | Current Position: President | Current Company: Institute for Defense AnalysesResume Parsing
Resume parsing is yet an even more complex example of information extraction.
!wget https://arun.maiya.net/asmcv.pdf -O /tmp/cv.pdf--2024-11-13 12:52:50-- https://arun.maiya.net/asmcv.pdf
Resolving arun.maiya.net (arun.maiya.net)... 185.199.109.153, 185.199.108.153, 185.199.111.153, ...
Connecting to arun.maiya.net (arun.maiya.net)|185.199.109.153|:443... connected.
HTTP request sent, awaiting response... 200 OK
Length: 62791 (61K) [application/pdf]
Saving to: ‘/tmp/cv.pdf’
/tmp/cv.pdf 100%[===================>] 61.32K --.-KB/s in 0.002s
2024-11-13 12:52:51 (36.3 MB/s) - ‘/tmp/cv.pdf’ saved [62791/62791]
from onprem.ingest import load_single_document
docs = load_single_document('/tmp/cv.pdf')
resume_text = docs[0].page_content # we'll only consider the first page of CV as "resume"prompt = """
Analyze the resume below and extract the relevant details. Format the response in JSON according to the specified structure below.
Only return the JSON response, with no additional text or explanations.
Ensure to:
- Format the full name in proper case.
- Remove any spaces and country code from the contact number.
- Format dates as "dd-mm-yyyy" if given in a more complex format, or retain the year if only the year is present.
- Do not make up a phone number.
- Extract only the first two jobs for Work Experience.
Use the following JSON structure:
```json
{
"Personal Information": {
"Name": " ",
"Contact Number": " ",
"Address": " ",
"Email": " ",
"Date of Birth": " "
},
"Education": [
{
"Degree": " ",
"Institution": " ",
"Year": " "
},
// Additional educational qualifications in a similar format
],
"Work Experience": [
{
"Position": " ",
"Organization": " ",
"Duration": " ",
"Responsibilities": " "
},
// Additional work experiences in a similar format
],
"Skills": [
{
"Skills": " ", // e.g., Python, R, Java, statistics, quantitative psychology, applied mathematics, machine learning, gel electrophoresis
},
// A list of skills or fields that the person has experience with
],
}
```
Here is the text of the resume:
---RESUMETXT---
"""json_string = llm.prompt(prompt.replace('---RESUMETXT---', resume_text)){
"Personal Information": {
"Name": "Arun S. Maiya",
"Contact Number": "",
"Address": "",
"Email": "arun@maiya.net",
"Date of Birth": ""
},
"Education": [
{
"Degree": "Ph.D.",
"Institution": "University of Illinois at Chicago",
"Year": " "
},
{
"Degree": "M.S.",
"Institution": "DePaul University",
"Year": " "
},
{
"Degree": "B.S.",
"Institution": "University of Illinois at Urbana-Champaign",
"Year": " "
}
],
"Work Experience": [
{
"Position": "Research Leader",
"Organization": "Institute for Defense Analyses – Alexandria, VA USA",
"Duration": "2011-Present",
"Responsibilities": ""
},
{
"Position": "Researcher",
"Organization": "University of Illinois at Chicago",
"Duration": "2007-2011",
"Responsibilities": ""
}
],
"Skills": [
{
"Skills": "applied machine learning, data science, natural language processing (NLP), network science, computer vision"
}
]
}Let’s convert the output to a Python dictionary:
import json
d = json.loads(json_string)
d.keys()dict_keys(['Personal Information', 'Education', 'Work Experience', 'Skills'])d['Personal Information']['Name']'Arun S. Maiya'Grammar Correction
prompt = """Here are some examples.
[Sentence]:
I love goin to the beach.
[Correction]: I love going to the beach.
[Sentence]:
Let me hav it!
[Correction]: Let me have it!
[Sentence]:
It have too many drawbacks.
[Correction]: It has too many drawbacks.
What is the correction for the following sentence?
[Sentence]:
I do not wan to go
[Correction]:"""
saved_output = llm.prompt(prompt, stop=[])I do not want to go.Classification
prompt = """Classify each sentence as either positive, negative, or neutral. Here are some examples.
[Sentence]: I love going to the beach.
[[Classification]: Positive
[Sentence]: It is 10am right now.
[Classification]: Neutral
[Sentence]: I just got fired from my job.
[Classification]: Negative
What is the classification for the following sentence? Answer with either Positive or Negative only.
[Sentence]: The reactivity of your team has been amazing, thanks!
[Classification]:"""
saved_output = llm.prompt(prompt, stop=['\n'])PositiveParaphrasing
prompt = """Paraphrase the following text delimited by triple backticks using a single sentence.
```After a war lasting 20 years, following the decision taken first by President Trump and then by President Biden to withdraw American troops, Kabul, the capital of Afghanistan, fell within a few hours to the Taliban, without resistance.```
"""
saved_output = llm.prompt(prompt)After a 20-year war, Kabul fell to the Taliban within hours after US troops withdrew under decisions made by Presidents Trump and Biden.Few-Shot Answer Extraction
prompt = """ Compelte the correct answer based on the Context. Answer should be a short word or phrase from Context.
[Question]: When was NLP Cloud founded?
[Context]: NLP Cloud was founded in 2021 when the team realized there was no easy way to reliably leverage Natural Language Processing in production.
[Answer]: 2021
[Question]: What did NLP Cloud develop?
[Context]: NLP Cloud developed their API by mid-2020 and they added many pre-trained open-source models since then.
[Answer]: API
[Question]: When can plans be stopped?
[Context]: All plans can be stopped anytime. You only pay for the time you used the service. In case of a downgrade, you will get a discount on your next invoice.
[Answer]: Anytime
[Question]: Which plan is recommended for GPT-J?
[Context]: The main challenge with GPT-J is memory consumption. Using a GPU plan is recommended.
[Answer]:"""
saved_output = llm.prompt(prompt, stop=['\n\n'])GPU planGenerating Product Descriptions
prompt = """Generate a short Sentence from the Keywords. Here are some examples.
[Keywords]: shoes, women, $59
[Sentence]: Beautiful shoes for women at the price of $59.
[Keywords]: trousers, men, $69
[Sentence]: Modern trousers for men, for $69 only.
[Keywords]: gloves, winter, $19
[Sentence]: Amazingly hot gloves for cold winters, at $19.
Generate a sentence for the following Keywords and nothing else:
[Keywords]: t-shirt, men, $39
[Sentence]:"""
saved_output = llm.prompt(prompt, stop=[])A comfortable t-shirt for men, available at $39.Tweet Generation
prompt = """Generate a tweet based on the supplied Keyword. Here are some examples.
[Keyword]:
markets
[Tweet]:
Take feedback from nature and markets, not from people
###
[Keyword]:
children
[Tweet]:
Maybe we die so we can come back as children.
###
[Keyword]:
startups
[Tweet]:
Startups should not worry about how to put out fires, they should worry about how to start them.
Generate a Tweet for the following keyword and nothing else:
###
[Keyword]:
climate change
[Tweet]:"""
saved_output = llm.prompt(prompt)The climate is not changing, it's us who are changing the climate.Generating an Email Draft
prompt = """Generate an email introducing Tesla to shareholders."""
saved_output = llm.prompt(prompt)Here is a draft email introducing Tesla to shareholders:
Subject: Welcome to Tesla, Inc.
Dear valued shareholder,
I am thrilled to introduce you to Tesla, Inc., the pioneering electric vehicle and clean energy company. As a shareholder, you are part of our mission to accelerate the world's transition to sustainable energy.
At Tesla, we are committed to pushing the boundaries of innovation and sustainability. Our products and services include:
* Electric vehicles: We design, manufacture, and sell electric vehicles that are not only environmentally friendly but also technologically advanced.
* Energy storage: Our energy storage products, such as the Powerwall and Powerpack, enable homeowners and businesses to store excess energy generated by their solar panels or other renewable sources.
* Solar energy: We design, manufacture, and install solar panel systems for residential and commercial customers.
As a shareholder, you are part of our journey towards a sustainable future. I invite you to explore our website and social media channels to learn more about our products and services.
Thank you for your support and trust in Tesla, Inc.
Sincerely,
[Your Name]
Tesla, Inc.
Note: This is just a draft email and may not be suitable for actual use.Talk to Your Documents
llm.ingest("./sample_data/")Appending to existing vectorstore at /home/amaiya/onprem_data/vectordb
Loading documents from ./sample_data/Loading new documents: 100%|██████████████████████| 1/1 [00:16<00:00, 16.09s/it]Loaded 1 new documents from ./sample_data/
Split into 12 chunks of text (max. 500 chars each)
Creating embeddings. May take some minutes...100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:00<00:00, 6.05it/s]Ingestion complete! You can now query your documents using the LLM.ask or LLM.chat methodsresult = llm.ask("What is ktrain?")Based on the provided context, ktrain is a tool that automates various aspects of the machine learning (ML) workflow. However, unlike traditional automation tools, ktrain also allows users to make choices and decisions that best fit their unique application requirements.
In essence, ktrain uses automation to augment and complement human engineers, rather than attempting to entirely replace them.Pro-Tip: You can try different models or re-phrase the question/prompts accordingly, which may provide better performance for certain tasks. For instance, by supplying default_model=zephyr to the LLM constructor and leaving model_url blank, the default Zephyr-7B-beta model will be used and also performs well on the above tasks. If not supplying any arguments to LLM, the default Mistral-7B-v0.2 model used, as shown in this example Google Colab notebook of OnPrem.LLM.