WatNeru commited on
Commit
52c76d4
·
1 Parent(s): 677f8d1

logの量を減らす

Browse files
Files changed (2) hide show
  1. app.py +4 -4
  2. package/rust_adapter.py +1 -1
app.py CHANGED
@@ -207,10 +207,10 @@ def build_word_tree(payload: WordTreeRequest) -> List[WordTreeResponse]:
207
 
208
  try:
209
  print(
210
- f"[API] build_word_tree called: prompt='{payload.prompt_text}', "
211
- f"root='{payload.root_text}', top_k={payload.top_k}, max_depth={payload.max_depth}"
212
  )
213
- print(f"[API] Adapter available: {adapter is not None}")
214
 
215
  results = adapter.build_word_tree(
216
  prompt_text=payload.prompt_text,
@@ -218,7 +218,7 @@ def build_word_tree(payload: WordTreeRequest) -> List[WordTreeResponse]:
218
  top_k=payload.top_k,
219
  max_depth=payload.max_depth,
220
  )
221
- print(f"[API] Generated {len(results)} candidates")
222
 
223
  if not results:
224
  print("[API] No candidates generated, returning dummy candidates")
 
207
 
208
  try:
209
  print(
210
+ f"[API] build_word_tree called: prompt=\n##########################\n{payload.prompt_text}\n##########################\n', "
211
+ f"root=\n%%%%%%%%%%%%%%%%%%%\n{payload.root_text}\n%%%%%%%%%%%%%%%%%%%\n', top_k={payload.top_k}, max_depth={payload.max_depth}"
212
  )
213
+ # print(f"[API] Adapter available: {adapter is not None}")
214
 
215
  results = adapter.build_word_tree(
216
  prompt_text=payload.prompt_text,
 
218
  top_k=payload.top_k,
219
  max_depth=payload.max_depth,
220
  )
221
+ # print(f"[API] Generated {len(results)} candidates")
222
 
223
  if not results:
224
  print("[API] No candidates generated, returning dummy candidates")
package/rust_adapter.py CHANGED
@@ -48,7 +48,7 @@ class RustAdapter:
48
  top_k=top_k,
49
  max_depth=max_depth,
50
  )
51
- print(f"[RustAdapter] build_word_tree: {pieces}")
52
  return [
53
  {"text": p.get_full_word(), "probability": float(p.probability)}
54
  for p in pieces
 
48
  top_k=top_k,
49
  max_depth=max_depth,
50
  )
51
+ # print(f"[RustAdapter] build_word_tree: {pieces}")
52
  return [
53
  {"text": p.get_full_word(), "probability": float(p.probability)}
54
  for p in pieces