langroid 0.42.1__tar.gz → 0.42.2__tar.gz

{langroid-0.42.1 → langroid-0.42.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langroid
-Version: 0.42.1
+Version: 0.42.2
 Summary: Harness LLMs with Multi-Agent Programming
 Author-email: Prasad Chalasani <pchalasani@gmail.com>
 License: MIT

{langroid-0.42.1 → langroid-0.42.2}/langroid/language_models/model_info.py

@@ -63,6 +63,7 @@ class GeminiModel(ModelName):
     GEMINI_1_5_FLASH = "gemini/gemini-1.5-flash"
     GEMINI_1_5_FLASH_8B = "gemini/gemini-1.5-flash-8b"
     GEMINI_1_5_PRO = "gemini/gemini-1.5-pro"
+    GEMINI_2_PRO = "gemini/gemini-2.0-pro-exp-02-05"
     GEMINI_2_FLASH = "gemini/gemini-2.0-flash"
     GEMINI_2_FLASH_LITE = "gemini/gemini-2.0-flash-lite-preview"
     GEMINI_2_FLASH_THINKING = "gemini/gemini-2.0-flash-thinking-exp"
@@ -322,6 +323,14 @@ MODEL_INFO: Dict[str, ModelInfo] = {
         rename_params={"max_tokens": "max_completion_tokens"},
         description="Gemini 1.5 Pro",
     ),
+    GeminiModel.GEMINI_2_PRO.value: ModelInfo(
+        name=GeminiModel.GEMINI_2_PRO.value,
+        provider=ModelProvider.GOOGLE,
+        context_length=2_000_000,
+        max_output_tokens=8192,
+        rename_params={"max_tokens": "max_completion_tokens"},
+        description="Gemini 2 Pro Exp 02-05",
+    ),
     GeminiModel.GEMINI_2_FLASH_THINKING.value: ModelInfo(
         name=GeminiModel.GEMINI_2_FLASH_THINKING.value,
         provider=ModelProvider.GOOGLE,

{langroid-0.42.1 → langroid-0.42.2}/langroid/prompts/templates.py

@@ -50,20 +50,23 @@ EXTRACTION_PROMPT = f"""
 
 SUMMARY_ANSWER_PROMPT_GPT4 = f"""
 
-        Use the provided NUMBERED extracts (with sources)  to answer the QUESTION. 
-        If there's not enough information, respond with {NO_ANSWER}. Use only the 
-        information in these extracts, even if your answer is factually incorrect, 
+        Use the provided NUMBERED EXTRACTS (with sources)  to answer the QUESTION. 
+        If there's not enough information, respond with {NO_ANSWER}. Use ONLY the 
+        information in these extracts, even if your answer is factually incorrect.
         and even if the answer contradicts other parts of the document. The only 
         important thing is that your answer is consistent with and supported by the 
         extracts. Compose your complete answer, inserting CITATIONS in MARKDOWN format
         [^i][^j] where i,j,... are the extract NUMBERS you are 
         citing.
-        For example your answer might look like this (NOTE HOW multiple citations
+        For EXAMPLE your answer might look like this (NOTE HOW multiple citations
         are grouped as [^2][^5]):
         
+        <ExampleAnswer>
         Beethoven composed the 9th symphony in 1824.[^1] After that he became deaf
         and could not hear his own music. [^2][^5]. He was a prolific composer and
         wrote many famous pieces.
+        </ExampleAnswer>
+        
         
         NUMBERED EXTRACTS:
         

{langroid-0.42.1 → langroid-0.42.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "langroid"
-version = "0.42.1"
+version = "0.42.2"
 authors = [
     {name = "Prasad Chalasani", email = "pchalasani@gmail.com"},
 ]