import os, sys, requests src_dir = os.path.join( os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) ) sys.path.append(src_dir) from configs.model_config import KB_ROOT_PATH, JUPYTER_WORK_PATH from configs.server_config import SANDBOX_SERVER from coagent.tools import toLangchainTools, TOOL_DICT, TOOL_SETS from coagent.llm_models.llm_config import EmbedConfig, LLMConfig from coagent.connector.phase import BasePhase from coagent.connector.schema import Message, Memory # log-level,print prompt和llm predict os.environ["log_verbose"] = "2" phase_name = "codeChatPhase" llm_config = LLMConfig( model_name="gpt-3.5-turbo", model_device="cpu",api_key=os.environ["OPENAI_API_KEY"], api_base_url=os.environ["API_BASE_URL"], temperature=0.3 ) embed_config = EmbedConfig( embed_engine="model", embed_model="text2vec-base-chinese", embed_model_path=os.path.join(src_dir, "embedding_models/text2vec-base-chinese") ) phase = BasePhase( phase_name, sandbox_server=SANDBOX_SERVER, jupyter_work_path=JUPYTER_WORK_PATH, embed_config=embed_config, llm_config=llm_config, kb_root_path=KB_ROOT_PATH, ) # 代码一共有多少类 => 基于cypher # 代码库里有哪些函数,返回5个就行 => 基于cypher # remove 这个函数是做什么的 => 基于标签 # 有没有函数已经实现了从字符串删除指定字符串的功能,使用的话可以怎么使用,写个java代码 => 基于描述 # 有根据我以下的需求用 java 开发一个方法:输入为字符串,将输入中的 .java 字符串给删除掉,然后返回新的字符串 => 基于描述 # round-1 query_content = "代码一共有多少类" query = Message( role_name="human", role_type="user", role_content=query_content, input_query=query_content, origin_query=query_content, code_engine_name="client", score_threshold=1.0, top_k=3, cb_search_type="cypher" ) output_message1, _ = phase.step(query) # round-2 query_content = "代码库里有哪些函数,返回5个就行" query = Message( role_name="human", role_type="user", role_content=query_content, input_query=query_content, origin_query=query_content, code_engine_name="client", score_threshold=1.0, top_k=3, cb_search_type="cypher" ) output_message2, _ = phase.step(query) # round-3 query_content = "remove 这个函数是做什么的" query = Message( role_name="user", role_type="human", role_content=query_content, input_query=query_content, origin_query=query_content, code_engine_name="client", score_threshold=1.0, top_k=3, cb_search_type="tag" ) output_message3, _ = phase.step(query) # round-4 query_content = "有没有函数已经实现了从字符串删除指定字符串的功能,使用的话可以怎么使用,写个java代码" query = Message( role_name="human", role_type="user", role_content=query_content, input_query=query_content, origin_query=query_content, code_engine_name="client", score_threshold=1.0, top_k=3, cb_search_type="description" ) output_message4, _ = phase.step(query) # round-5 query_content = "有根据我以下的需求用 java 开发一个方法:输入为字符串,将输入中的 .java 字符串给删除掉,然后返回新的字符串" query = Message( role_name="human", role_type="user", role_content=query_content, input_query=query_content, origin_query=query_content, code_engine_name="client", score_threshold=1.0, top_k=3, cb_search_type="description" ) output_message5, output_memory5 = phase.step(query) print(output_memory5.to_str_messages(return_all=True, content_key="parsed_output_list"))