파이썬 - 리눅스에서 GPT 명령어 사용하기

 

모듈 설치

pip install openai

 

~/dev/gpt.py

import openai
import sys

openai.api_key = "자신의 API 토큰을 여기에 입력"

messages = [
	{"role": "system", "content": "You are a helpful assistant."},
]

def request(text:str):

	global messages

	if len(messages) >= 30:
		messages = messages[-10:]

	query = text
	messages.append({"role": "user", "content": query})

	response = openai.ChatCompletion.create(
		model="gpt-3.5-turbo",
		messages=messages,
		stream=True,
		max_tokens=255
	)

	answer = ""

	for chunk in response:
		if "content" in chunk['choices'][0]['delta']:
			word = chunk['choices'][0]['delta']['content']
			print(word, end="")

request(" ".join(sys.argv[1:]))

 

bash 환경변수 등록

$ nano ~/.bashrc

# 맨 하단에 아래 명령어 추가
alias gpt="python3 ~/dev/gpt.py"

$ source ~/.bashrc

 

 

사용 방법

$ gpt 파이썬으로 hello world를 출력을 하려면?
다음과 같이 코드를 작성하세요.

```
print("hello world")
```

이 코드를 실행하면 "hello world" 라는 메시지가 출력됩니다.