Depedency fixes

Adding more arg support to GeneratorParams
dreadnode · Feb 2, 2024 · b5f472b · b5f472b
1 parent 10f6af5
commit b5f472b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rigging"
-version = "0.1.2"
+version = "0.1.3"
 description = "LLM Interaction Framework"
 authors = ["Nick Landers <monoxgas@gmail.com>"]
 license = "MIT"
@@ -9,8 +9,8 @@ readme = "README.md"
 
 [tool.poetry.dependencies]
 python = "<3.13,>=3.10"
-pydantic = "^2.5.3"
-pydantic-xml = "^2.7.0"
+pydantic = "2.5.3"
+pydantic-xml = "2.7.0"
 loguru = "^0.7.2"
 litellm = "^1.20.1"
 

diff --git a/rigging/generator.py b/rigging/generator.py
@@ -34,6 +34,10 @@ class GenerateParams(BaseModel):
     stop: list[str] | None = None
     presence_penalty: float | None = None
     frequency_penalty: float | None = None
+    api_key: str | None = None
+    api_base: str | None = None
+    timeout: int | None = None
+    seed: int | None = None
 
     @field_validator("stop", mode="before")
     def validate_stop(cls, value: t.Any) -> t.Any: