(no commit message)

This commit is contained in:
2026-02-04 21:38:54 -08:00
parent ac32b3fd42
commit 01421613da
19 changed files with 107 additions and 2 deletions

View File

@@ -1,2 +0,0 @@
# nested_repo_3

5
auto_classes.json Normal file
View File

@@ -0,0 +1,5 @@
{
"AutoConfig": "program.config.AgentWRetreiverConfig",
"AutoProgram": "program.program.AgentWRetreiver",
"AutoRetriever": "program.retriever.ExampleRetriever"
}

15
config.json Normal file
View File

@@ -0,0 +1,15 @@
{
"model": null,
"num_fetch": 1,
"lm": "openai/gpt-4o-mini",
"embedder": "openai/text-embedding-3-small",
"clients": {
"mit": [
"csail",
"mit-media-lab"
],
"berkeley": [
"bear"
]
}
}

9
program.json Normal file
View File

@@ -0,0 +1,9 @@
{
"metadata": {
"dependency_versions": {
"python": "3.13",
"dspy": "3.1.2",
"cloudpickle": "3.1"
}
}
}

0
program/__init__.py Normal file
View File

10
program/config.py Normal file
View File

@@ -0,0 +1,10 @@
from pydantic import Field
from modaic import PrecompiledConfig
class AgentWRetreiverConfig(PrecompiledConfig):
num_fetch: int
lm: str = "openai/gpt-4o-mini"
embedder: str = "openai/text-embedding-3-small"
clients: dict = Field(default_factory=lambda: {"mit": ["csail", "mit-media-lab"], "berkeley": ["bear"]})

29
program/program.py Normal file
View File

@@ -0,0 +1,29 @@
import sys
from modaic import PrecompiledProgram
from .config import AgentWRetreiverConfig
from .retriever import ExampleRetriever
from .tools.google.google_search import search_google # noqa: F401
from .utils.used import random_util # noqa: F401
class AgentWRetreiver(PrecompiledProgram):
config: AgentWRetreiverConfig
def __init__(self, config: AgentWRetreiverConfig, retriever: ExampleRetriever, **kwargs):
super().__init__(config, retriever=retriever, **kwargs)
self.lm = self.config.lm
self.clients = self.config.clients
def forward(self, query: str) -> str:
return self.retriever.retrieve(query)
if __name__ == "__main__":
username = sys.argv[1] # ← first arg after script name (username)
config = AgentWRetreiverConfig(num_fetch=1)
retriever = ExampleRetriever(config, needed_param="hi")
program = AgentWRetreiver(config, retriever=retriever)
repo_path = f"{username}/nested_repo_3"
program.push_to_hub(repo_path, with_code=True)

17
program/retriever.py Normal file
View File

@@ -0,0 +1,17 @@
from modaic import Retriever
from .config import AgentWRetreiverConfig
from .tools.jira.jira_api_tools import call_jira_api # noqa: F401
from .utils.used import random_util # noqa: F401
class ExampleRetriever(Retriever):
config: AgentWRetreiverConfig
def __init__(self, config: AgentWRetreiverConfig, needed_param: str, **kwargs):
super().__init__(config, **kwargs)
self.embedder_name = config.embedder
self.needed_param = needed_param
def retrieve(self, query: str) -> str:
return f"Retrieved {self.config.num_fetch} results for {query}"

View File

View File

View File

@@ -0,0 +1,2 @@
def search_google() -> str:
return "search_google"

View File

View File

@@ -0,0 +1,2 @@
def call_jira_api() -> str:
return "call_jira_api"

View File

View File

View File

@@ -0,0 +1,2 @@
def import_me_too() -> str:
return "import_me_too"

View File

5
program/utils/used.py Normal file
View File

@@ -0,0 +1,5 @@
from .second_degree_import import import_me_too # noqa: F401
def random_util() -> str:
return "random_util"

11
pyproject.toml Normal file
View File

@@ -0,0 +1,11 @@
[project]
name = "nested_repo_3"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.10"
dependencies = ["dspy", "modaic"]
[tool.uv.sources]
modaic = { workspace = true }