Upload 3 files
Browse files- config.json +1 -1
- emotion_filter.py +0 -1
- external_api_engine.py +4 -8
config.json
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
{
|
2 |
"engine": "external"
|
3 |
-
}
|
|
|
1 |
{
|
2 |
"engine": "external"
|
3 |
+
}
|
emotion_filter.py
CHANGED
@@ -16,5 +16,4 @@ def rewrite_with_persona(raw_response, emotion="neutral", relation="trusted"):
|
|
16 |
suffix = suffix_options.get(emotion, suffix_options["neutral"])
|
17 |
response = raw_response.strip()
|
18 |
|
19 |
-
# 整形
|
20 |
return f"{prefix}{response}{suffix}"
|
|
|
16 |
suffix = suffix_options.get(emotion, suffix_options["neutral"])
|
17 |
response = raw_response.strip()
|
18 |
|
|
|
19 |
return f"{prefix}{response}{suffix}"
|
external_api_engine.py
CHANGED
@@ -1,8 +1,9 @@
|
|
1 |
|
2 |
import requests
|
3 |
import os
|
|
|
4 |
|
5 |
-
API_TOKEN = os.getenv("neko")
|
6 |
API_URL = "https://api-inference.huggingface.co/models/rinna/japanese-gpt-neox-3.6b"
|
7 |
|
8 |
headers = {
|
@@ -24,14 +25,9 @@ def call_external_api(input_text):
|
|
24 |
generated = response.json()
|
25 |
|
26 |
if isinstance(generated, list) and "generated_text" in generated[0]:
|
27 |
-
|
|
|
28 |
else:
|
29 |
return "[API応答形式が不明です]"
|
30 |
except Exception as e:
|
31 |
return f"[Hugging Face APIエラー] {str(e)}"
|
32 |
-
from emotion_filter import rewrite_with_persona
|
33 |
-
|
34 |
-
# API応答をけいすけ語に整形
|
35 |
-
api_response = call_external_api(user_input)
|
36 |
-
keisuke_response = rewrite_with_persona(api_response, emotion="gentle")
|
37 |
-
return keisuke_response
|
|
|
1 |
|
2 |
import requests
|
3 |
import os
|
4 |
+
from emotion_filter import rewrite_with_persona
|
5 |
|
6 |
+
API_TOKEN = os.getenv("neko")
|
7 |
API_URL = "https://api-inference.huggingface.co/models/rinna/japanese-gpt-neox-3.6b"
|
8 |
|
9 |
headers = {
|
|
|
25 |
generated = response.json()
|
26 |
|
27 |
if isinstance(generated, list) and "generated_text" in generated[0]:
|
28 |
+
raw_text = generated[0]["generated_text"]
|
29 |
+
return rewrite_with_persona(raw_text, emotion="gentle")
|
30 |
else:
|
31 |
return "[API応答形式が不明です]"
|
32 |
except Exception as e:
|
33 |
return f"[Hugging Face APIエラー] {str(e)}"
|
|
|
|
|
|
|
|
|
|
|
|