Adapters
Korean
English
gabeen commited on
Commit
1599016
·
1 Parent(s): 455edbb

Upload ai.py

Browse files
Files changed (1) hide show
  1. ai.py +49 -0
ai.py ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import openai
2
+ # OpenAI API 인증 정보 설정
3
+ openai.api_key = "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"
4
+ #openai.Model.list()
5
+
6
+ # import history_message
7
+
8
+ print("반려견 헬스케어 AI입니다. 무엇을 도와드릴까요?")
9
+
10
+ # 이전 대화 내용을 저장할 리스트
11
+ history_message = [
12
+ {"role": "system", "content": "Companion Pet Healthcare AI Assistant"}
13
+ ]
14
+ # GPT-3 엔진 선택
15
+ model_engine = "gpt-3.5-turbo"
16
+
17
+ # OpenAI API를 호출하여 대화를 생성하는 함수
18
+ def generate_chat(question):
19
+ # OpenAI API 호출하여 대화 생성
20
+ history_message.append({"role":"user", "content":question})
21
+ print(history_message)
22
+ completions = openai.ChatCompletion.create(
23
+ model="gpt-3.5-turbo",
24
+ messages=history_message
25
+ )
26
+ message = completions.choices[0].message.to_dict()
27
+ print(message)
28
+ answer = message["content"].strip()
29
+
30
+ # 이전 대화 내용에 새로운 답변 추가
31
+ history_message.append(message)
32
+ print(history_message)
33
+ return answer
34
+
35
+ # 사용자와의 대화 반복
36
+ while True:
37
+ # 사용자의 입력 받기
38
+ question = input("User: ")
39
+
40
+ # 대화 종료 조건 확인
41
+ if question.lower() in ["exit", "quit", "goodbye"]:
42
+ print("AI: 감사합니다.")
43
+ break
44
+
45
+ # OpenAI API 호출하여 답변 생성
46
+ answer = generate_chat(question)
47
+
48
+ # 챗봇의 답변 출력
49
+ print("AI:", answer)