You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70 lines
1.9 KiB
Markdown
70 lines
1.9 KiB
Markdown
# 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) |