Skip to content

Commit 974ad35

Browse files
author
shuang
committed
add missing import
1 parent 91cd319 commit 974ad35

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

gptbioinsightor/utils.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,9 @@ def close(self):
9393

9494

9595
def get_pre_celltype_chat(cluster_num, background, provider, model, base_url, sys_prompt):
96+
from .core import query_model
97+
from .prompt import PRE_CELLTYPE_PROMPT
98+
from concurrent.futures import ThreadPoolExecutor
9699
text = PRE_CELLTYPE_PROMPT.format(number=cluster_num, background=background)
97100
msg = [{"role": "user", "content": text}]
98101
from functools import partial

0 commit comments

Comments
 (0)