mirror of
https://github.com/dataforcanada/d4c-service-geo-assistant.git
synced 2026-06-14 23:11:01 +02:00
Add docs to download overture places data & ollama model (#13)
* Add docs to download overture places data & ollama model * Hit local overture parquet files * Add osx gitignore * Add .env.example * Make overture data source selectable using .env * Add pytest marker to set right ENV vars during CI --------- Co-authored-by: Daniel Wiesmann <yellowcap@users.noreply.github.com>
This commit is contained in:
committed by
GitHub
parent
16f39f908f
commit
be8affaa6c
@@ -1,5 +1,6 @@
|
||||
"""Tools for summarizing satellite images using LLM-based analysis."""
|
||||
|
||||
import os
|
||||
from typing import Annotated, Optional
|
||||
import dspy
|
||||
from langchain_core.tools import tool
|
||||
@@ -7,6 +8,10 @@ from langgraph.types import Command
|
||||
from langchain_core.messages import ToolMessage
|
||||
from langchain_core.tools.base import InjectedToolCallId
|
||||
|
||||
import dotenv
|
||||
|
||||
dotenv.load_dotenv()
|
||||
|
||||
|
||||
class SatImgSummary(dspy.Signature):
|
||||
"Describe things you see in the satellite image."
|
||||
@@ -20,8 +25,8 @@ class SatImgSummaryAgent(dspy.Module):
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
model: str = "ministral-3:14b-cloud",
|
||||
api_base: str = "http://localhost:11434",
|
||||
model: str = os.environ.get("OLLAMA_IMAGE_MODEL", "ministral-3:14b-cloud"),
|
||||
api_base: str = os.environ.get("OLLAMA_BASE_URL", "http://localhost:11434"),
|
||||
temperature: float = 0.5,
|
||||
max_tokens: int = 4_096,
|
||||
) -> None:
|
||||
|
||||
Reference in New Issue
Block a user