DeL-TaiseiOzaki
commited on
Commit
·
443a8ae
1
Parent(s):
9477a09
- services/llm_service.py +8 -3
services/llm_service.py
CHANGED
@@ -48,16 +48,21 @@ class LLMService:
|
|
48 |
self.conversation_history = []
|
49 |
|
50 |
def get_response(self, content: str, query: str) -> tuple[Optional[str], Optional[str]]:
|
|
|
51 |
try:
|
52 |
prompt = self.create_prompt(content, query)
|
53 |
self._add_to_history("user", prompt)
|
54 |
|
|
|
55 |
response = self.claude_client.messages.create(
|
56 |
-
model="claude-3-
|
57 |
-
|
|
|
|
|
|
|
58 |
)
|
59 |
-
answer = response.content[0].text
|
60 |
|
|
|
61 |
self._add_to_history("assistant", answer)
|
62 |
return answer, None
|
63 |
|
|
|
48 |
self.conversation_history = []
|
49 |
|
50 |
def get_response(self, content: str, query: str) -> tuple[Optional[str], Optional[str]]:
|
51 |
+
"""LLMを使用して回答を生成"""
|
52 |
try:
|
53 |
prompt = self.create_prompt(content, query)
|
54 |
self._add_to_history("user", prompt)
|
55 |
|
56 |
+
# Claudeへのリクエストを修正
|
57 |
response = self.claude_client.messages.create(
|
58 |
+
model="claude-3-sonnet-20240229",
|
59 |
+
max_tokens=4096,
|
60 |
+
messages=[
|
61 |
+
{"role": "user", "content": prompt}
|
62 |
+
]
|
63 |
)
|
|
|
64 |
|
65 |
+
answer = response.content[0].text
|
66 |
self._add_to_history("assistant", answer)
|
67 |
return answer, None
|
68 |
|