|
22 | 22 | MISTRAL_API_URL = "https://api.mistral.ai/v1/chat/completions"
|
23 | 23 |
|
24 | 24 | def create_translation_prompt(target_language, special=False):
|
| 25 | + base_prompt = "You are translating a markdown file for a Jekyll blog post. Be careful about code blocks. Translate to {target_language}. Just give translated post.\n\n" |
25 | 26 | if target_language == 'ja':
|
26 |
| - return "Translate to Japanese. Just give translated text.\n\n" |
| 27 | + return base_prompt.format(target_language="Japanese") |
27 | 28 | elif target_language == 'es':
|
28 |
| - return "Translate to Spanish. Just give translated text.\n\n" |
| 29 | + return base_prompt.format(target_language="Spanish") |
29 | 30 | elif target_language == 'hi':
|
30 |
| - return "Translate to Hindi. Just give translated text.\n\n" |
| 31 | + return base_prompt.format(target_language="Hindi") |
31 | 32 | elif target_language == 'fr':
|
32 |
| - return "Translate to French. Just give translated text.\n\n" |
| 33 | + return base_prompt.format(target_language="French") |
33 | 34 | elif target_language == "zh":
|
34 |
| - return f"""Translate to Chinese. Just give translated text.\n\n""" |
| 35 | + return base_prompt.format(target_language="Chinese") |
35 | 36 | elif target_language == 'hant':
|
36 |
| - return "Translate to Traditional Chinese (Hong Kong). Just give translated text.\n\n" |
| 37 | + return base_prompt.format(target_language="Traditional Chinese (Hong Kong)") |
37 | 38 | elif target_language == 'en':
|
38 |
| - return "Translate to English. Just give translated text.\n\n" |
| 39 | + return base_prompt.format(target_language="English") |
39 | 40 | elif target_language == 'de':
|
40 |
| - return "Translate to German. Just give translated text.\n\n" |
| 41 | + return base_prompt.format(target_language="German") |
41 | 42 | elif target_language == 'ar':
|
42 |
| - return "Translate to Arabic. Just give translated text.\n\n" |
| 43 | + return base_prompt.format(target_language="Arabic") |
43 | 44 | else:
|
44 |
| - return f"Translate to {target_language}. Just give translated text.\n\n" |
| 45 | + return base_prompt.format(target_language=target_language) |
45 | 46 |
|
46 | 47 | def call_mistral_api(prompt):
|
47 | 48 | api_key = MISTRAL_API_KEY
|
|
0 commit comments