(gpt_prompt, user_prompt, type_name, model_code, user_key="", base_url="default_url",
model: str = None, process_response_func=None)
| 45 | |
| 46 | |
| 47 | async def get_code(gpt_prompt, user_prompt, type_name, model_code, user_key="", base_url="default_url", |
| 48 | model: str = None, process_response_func=None): |
| 49 | new_prompt = prompt_type_buffer(gpt_prompt, user_prompt, model_code) |
| 50 | |
| 51 | if type_name == "glm": |
| 52 | # URL、headers和data |
| 53 | url, headers, data = configure_for_glm(base_url, model, user_key, new_prompt) |
| 54 | elif type_name == "gpt": |
| 55 | url, headers, data = configure_for_gpt(base_url, model, user_key, new_prompt) |
| 56 | else: |
| 57 | raise ValueError("Unsupported model type") |
| 58 | |
| 59 | return await send_request_and_process_response(url, headers, data, type_name, process_response_func or extract_code) |
| 60 | |
| 61 | |
| 62 | def add_prompt_type_buffer(user_prompt, model_code): |
no test coverage detected