| import json |
| from typing import Dict, Any, Optional |
|
|
| def estimate_tokens(text: str) -> int: |
| if not text: |
| return 0 |
| return int(len(text) / 4) + 1 |
|
|
| def format_sse_chunk(content: str, done: bool = False, usage: Optional[Dict[str, Any]] = None) -> str: |
| data = {"content": content, "done": done, "usage": usage} |
| return f"data: {json.dumps(data)}\n\n" |
|
|
| def get_completion_prompt(prefix: str, suffix: str, language: str) -> str: |
| return ( |
| f"Continue the code for the {language} language. " |
| f"Fill in the missing code between the Prefix and Suffix. " |
| f"Return ONLY the code to insert. No explanations, no markdown.\n\n" |
| f"--- PREFIX ---\n{prefix}\n" |
| f"--- SUFFIX ---\n{suffix}\n" |
| "--- INSERT COMPLETED CODE BELOW ---" |
| ) |
|
|
