rusinopp commited on
Commit
a158524
·
verified ·
1 Parent(s): 334aae4

Upload 3 files

Browse files
Files changed (3) hide show
  1. config.json +1 -1
  2. emotion_filter.py +0 -1
  3. 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") # ← Hugging Face Secrets に '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
- return generated[0]["generated_text"]
 
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)}"