mirror of
https://github.com/dair-ai/Prompt-Engineering-Guide
synced 2024-11-08 07:10:41 +00:00
70 lines
1.9 KiB
Plaintext
70 lines
1.9 KiB
Plaintext
|
# Generate Code Snippets with LLMs
|
||
|
|
||
|
import { Tabs, Tab } from 'nextra/components'
|
||
|
|
||
|
## Background
|
||
|
This prompt tests an LLM's code generation capabilities by prompting it to generate the corresponding code snippet given details about the program through a comment using `/* <instruction> */`.
|
||
|
|
||
|
## Prompt
|
||
|
```markdown
|
||
|
/*
|
||
|
Ask the user for their name and say "Hello"
|
||
|
*/
|
||
|
```
|
||
|
|
||
|
## Code / API
|
||
|
|
||
|
<Tabs items={['GPT-4 (OpenAI)', 'Mixtral MoE 8x7B Instruct (Fireworks)']}>
|
||
|
<Tab>
|
||
|
```python
|
||
|
from openai import OpenAI
|
||
|
client = OpenAI()
|
||
|
|
||
|
response = client.chat.completions.create(
|
||
|
model="gpt-4",
|
||
|
messages=[
|
||
|
{
|
||
|
"role": "user",
|
||
|
"content": "/*\nAsk the user for their name and say \"Hello\"\n*/"
|
||
|
}
|
||
|
],
|
||
|
temperature=1,
|
||
|
max_tokens=1000,
|
||
|
top_p=1,
|
||
|
frequency_penalty=0,
|
||
|
presence_penalty=0
|
||
|
)
|
||
|
```
|
||
|
</Tab>
|
||
|
|
||
|
<Tab>
|
||
|
```python
|
||
|
import fireworks.client
|
||
|
fireworks.client.api_key = "<FIREWORKS_API_KEY>"
|
||
|
completion = fireworks.client.ChatCompletion.create(
|
||
|
model="accounts/fireworks/models/mixtral-8x7b-instruct",
|
||
|
messages=[
|
||
|
{
|
||
|
"role": "user",
|
||
|
"content": "/*\nAsk the user for their name and say \"Hello\"\n*/",
|
||
|
}
|
||
|
],
|
||
|
stop=["<|im_start|>","<|im_end|>","<|endoftext|>"],
|
||
|
stream=True,
|
||
|
n=1,
|
||
|
top_p=1,
|
||
|
top_k=40,
|
||
|
presence_penalty=0,
|
||
|
frequency_penalty=0,
|
||
|
prompt_truncate_len=1024,
|
||
|
context_length_exceeded_behavior="truncate",
|
||
|
temperature=0.9,
|
||
|
max_tokens=4000
|
||
|
)
|
||
|
```
|
||
|
</Tab>
|
||
|
</Tabs>
|
||
|
|
||
|
|
||
|
## Reference
|
||
|
- [Prompt Engineering Guide](https://www.promptingguide.ai/introduction/examples#code-generation) (16 March 2023)
|