AI finance TA writer, yfinance, pandas_ta, WIP

2024-05-16 08:55:33 +05:30
parent 45508d318b
commit e3b51e2713
2 changed files with 9 additions and 9 deletions
--- a/lib/gpt_providers/text_generation/gemini_pro_text.py
+++ b/lib/gpt_providers/text_generation/gemini_pro_text.py
@@ -36,7 +36,7 @@ def gemini_text_response(prompt, temperature, top_p, n, max_tokens):
        "max_output_tokens": max_tokens,
    }
    # FIXME: Expose model_name in main_config
-    model = genai.GenerativeModel(model_name="gemini-pro", generation_config=generation_config)
+    model = genai.GenerativeModel(model_name="gemini-1.5-pro-latest", generation_config=generation_config)
    try:
        # text_response = []
        response = model.generate_content(prompt, stream=True)