sourcefire 0.3.0__tar.gz → 0.3.1__tar.gz

{sourcefire-0.3.0/sourcefire.egg-info → sourcefire-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sourcefire
-Version: 0.3.0
+Version: 0.3.1
 Summary: Get instant context on any codebase. One command to index, ask questions in plain English, get answers grounded in actual code.
 Author-email: Athar Wani <athar@cravv.com>
 License: MIT

{sourcefire-0.3.0 → sourcefire-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sourcefire"
-version = "0.3.0"
+version = "0.3.1"
 description = "Get instant context on any codebase. One command to index, ask questions in plain English, get answers grounded in actual code."
 readme = "README.md"
 license = {text = "MIT"}

{sourcefire-0.3.0 → sourcefire-0.3.1}/sourcefire/api/models.py

@@ -8,7 +8,7 @@ from pydantic import BaseModel
 class QueryRequest(BaseModel):
     query: str
     mode: Literal["debug", "feature", "explain"] = "debug"
-    model: Literal["gemini-2.5-flash", "gemini-2.5-pro"] = "gemini-2.5-flash"
+    model: Literal["gemini-3.1-flash-lite-preview", "gemini-3.1-pro-preview"] = "gemini-3.1-flash-lite-preview"
     history: list[dict] = []
 
 

{sourcefire-0.3.0 → sourcefire-0.3.1}/sourcefire/chain/rag_chain.py

@@ -827,7 +827,7 @@ async def stream_rag_response(
     graph: ImportGraph,
     query: str,
     mode: str,
-    model: str = "gemini-2.5-flash",
+    model: str = "gemini-3.1-flash-lite-preview",
     history: list[dict[str, str]] | None = None,
     profile: LanguageProfile | None = None,
     project_dir: Path | None = None,

{sourcefire-0.3.0 → sourcefire-0.3.1}/sourcefire/config.py

@@ -30,7 +30,7 @@ class SourcefireConfig:
 
     # [llm]
     provider: str = "gemini"
-    model: str = "gemini-2.5-flash"
+    model: str = "gemini-3.1-flash-lite-preview"
     api_key_env: str = "GEMINI_API_KEY"
 
     # [server]
@@ -68,8 +68,8 @@ class SourcefireConfig:
 # Constants used by other modules
 EMBEDDING_MODEL: str = "sentence-transformers/all-MiniLM-L6-v2"
 MAX_TOKEN_BUDGET: dict[str, int] = {
-    "gemini-2.5-flash": 100_000,
-    "gemini-2.5-pro": 200_000,
+    "gemini-3.1-flash-lite-preview": 100_000,
+    "gemini-3.1-pro-preview": 200_000,
 }
 MAX_HISTORY_PAIRS: int = 5
 RESPONSE_HEADROOM: int = 8_000
@@ -118,7 +118,7 @@ def load_config(project_dir: Path, sourcefire_dir: Path) -> SourcefireConfig:
         chunk_size=indexer.get("chunk_size", 1000),
         chunk_overlap=indexer.get("chunk_overlap", 300),
         provider=llm.get("provider", "gemini"),
-        model=llm.get("model", "gemini-2.5-flash"),
+        model=llm.get("model", "gemini-3.1-flash-lite-preview"),
         api_key_env=llm.get("api_key_env", "GEMINI_API_KEY"),
         host=server.get("host", "127.0.0.1"),
         port=server.get("port", 8000),

{sourcefire-0.3.0 → sourcefire-0.3.1}/sourcefire/init.py

@@ -49,7 +49,7 @@ def _generate_patterns_via_llm(file_tree: str, api_key: str) -> dict[str, list[s
         from langchain_core.messages import HumanMessage
 
         llm = ChatGoogleGenerativeAI(
-            model="gemini-2.5-flash",
+            model="gemini-3.1-flash-lite-preview",
             google_api_key=api_key,
         )
 

{sourcefire-0.3.0 → sourcefire-0.3.1/sourcefire.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sourcefire
-Version: 0.3.0
+Version: 0.3.1
 Summary: Get instant context on any codebase. One command to index, ask questions in plain English, get answers grounded in actual code.
 Author-email: Athar Wani <athar@cravv.com>
 License: MIT