kenken999 commited on
Commit
370ab91
1 Parent(s): 117e8d0
Files changed (2) hide show
  1. babyagi/babyagi.py +4 -4
  2. mysite/routers/fastapi.py +11 -0
babyagi/babyagi.py CHANGED
@@ -166,11 +166,11 @@ thread_name = ""
166
  args = sys.argv
167
  if len(args) > 1:
168
  print(args[1])
169
- OBJECTIVE = args[1]
170
  # ファイルを開いて内容を読み込む
171
- with open('/home/user/app/babyagi/prompt.txt', 'r') as file:
172
- data = file.read()
173
- thread_name = args[2]
174
 
175
  else:
176
  print("not args")
 
166
  args = sys.argv
167
  if len(args) > 1:
168
  print(args[1])
169
+ thread_name = args[1]
170
  # ファイルを開いて内容を読み込む
171
+ #with open('/home/user/app/babyagi/prompt.txt', 'r') as file:
172
+ # data = file.read()
173
+ #thread_name = args[2]
174
 
175
  else:
176
  print("not args")
mysite/routers/fastapi.py CHANGED
@@ -334,6 +334,17 @@ def setup_webhook_routes(app: FastAPI):
334
  # コマンドを構築
335
  command = f"""make runbabyagi "{text}に対して、より良いチャットボットでのQAプランデータ設定の提案を日本語で作成してください" {thread_name}"""
336
 
 
 
 
 
 
 
 
 
 
 
 
337
  """
338
  proc = subprocess.Popen(
339
  command,
 
334
  # コマンドを構築
335
  command = f"""make runbabyagi "{text}に対して、より良いチャットボットでのQAプランデータ設定の提案を日本語で作成してください" {thread_name}"""
336
 
337
+ with open('/home/user/app/babyagi/prompt.txt', 'w') as file:
338
+ file.write(f"""{text}の質問 についてチャットボットでよりよく対応するプラン""")
339
+
340
+
341
+ proc = subprocess.Popen(
342
+ ["make", "runbabyagi", thread_name],
343
+ stdin=subprocess.PIPE,
344
+ stdout=subprocess.PIPE,
345
+ stderr=subprocess.PIPE,
346
+ text=True,
347
+ )
348
  """
349
  proc = subprocess.Popen(
350
  command,