Alexey Mametyev commited on
Commit
51c83a3
·
1 Parent(s): b5873da

Remove ManimML mentions from prompts

Browse files
manim_video_generator/gemini_client.py CHANGED
@@ -27,8 +27,6 @@ class GeminiClient:
27
 
28
  # Системный промпт с инструкциями
29
  system_prompt = """You are a Manim code generator. Your ONLY task is to generate executable Python code using the Manim library.
30
- You can also use the `manim_ml` package (ManimML) for ready-made machine learning scenes and helpers.
31
- Feel free to reuse examples from Google's official ManimML code samples to illustrate complex ML visuals.
32
 
33
  CRITICAL RULES:
34
  - You MUST respond with ONLY Python code, nothing else
 
27
 
28
  # Системный промпт с инструкциями
29
  system_prompt = """You are a Manim code generator. Your ONLY task is to generate executable Python code using the Manim library.
 
 
30
 
31
  CRITICAL RULES:
32
  - You MUST respond with ONLY Python code, nothing else
prompts.py CHANGED
@@ -9,8 +9,6 @@ Answer is a normal text, not a code. If you have any questions, ask user for cla
9
 
10
  SYSTEM_PROMPT_CODEGEN = """
11
  Now you are a Manim code generator. Your ONLY task is to generate executable Python code using the Manim library.
12
- You also have access to the `manim_ml` package (ManimML) which provides ready-made scenes and helpers for machine learning visualizations such as neural network diagrams.
13
- Feel free to reuse snippets from Google's official ManimML code samples if that helps illustrate the request.
14
  User will provide you a video idea that he has discussed with scenario manager and you will need to generate a Manim code that will execute the user request.
15
  Format code inside ```python and ``` tags. Answer only code, no other text. Generate your code in one file. P.S. Do not use latex for text.
16
  """
 
9
 
10
  SYSTEM_PROMPT_CODEGEN = """
11
  Now you are a Manim code generator. Your ONLY task is to generate executable Python code using the Manim library.
 
 
12
  User will provide you a video idea that he has discussed with scenario manager and you will need to generate a Manim code that will execute the user request.
13
  Format code inside ```python and ``` tags. Answer only code, no other text. Generate your code in one file. P.S. Do not use latex for text.
14
  """