tzamuncode 0.1.2__tar.gz → 0.1.3__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {tzamuncode-0.1.2/tzamuncode.egg-info → tzamuncode-0.1.3}/PKG-INFO +1 -1
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/pyproject.toml +1 -1
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/models/ollama.py +5 -1
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/models/vllm_client.py +5 -1
- {tzamuncode-0.1.2 → tzamuncode-0.1.3/tzamuncode.egg-info}/PKG-INFO +1 -1
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/LICENSE +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/README.md +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/setup.cfg +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/__init__.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/agents/__init__.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/agents/coder.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/agents/tools.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/auth/__init__.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/auth/auth_manager.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/cli/__init__.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/cli/agentic_commands.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/cli/auth_commands.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/cli/commands.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/cli/enhanced_chat.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/cli/interactive_chat.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/cli/main.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/cli/realtime_chat.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/cli/realtime_chat_methods.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/cli/tui_chat.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/config/__init__.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/models/__init__.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/utils/__init__.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/utils/file_ops.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode/utils/project_scanner.py +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode.egg-info/SOURCES.txt +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode.egg-info/dependency_links.txt +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode.egg-info/entry_points.txt +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode.egg-info/requires.txt +0 -0
- {tzamuncode-0.1.2 → tzamuncode-0.1.3}/tzamuncode.egg-info/top_level.txt +0 -0
|
@@ -3,6 +3,7 @@ Ollama client for TzamunCode
|
|
|
3
3
|
"""
|
|
4
4
|
|
|
5
5
|
import requests
|
|
6
|
+
import os
|
|
6
7
|
from typing import List, Dict, Generator, Optional
|
|
7
8
|
import json
|
|
8
9
|
|
|
@@ -11,10 +12,13 @@ class OllamaClient:
|
|
|
11
12
|
|
|
12
13
|
def __init__(
|
|
13
14
|
self,
|
|
14
|
-
base_url: str =
|
|
15
|
+
base_url: str = None,
|
|
15
16
|
model: str = "qwen2.5:32b",
|
|
16
17
|
timeout: int = 120
|
|
17
18
|
):
|
|
19
|
+
# Use environment variable if available, otherwise default to localhost
|
|
20
|
+
if base_url is None:
|
|
21
|
+
base_url = os.getenv('OLLAMA_BASE_URL', 'http://localhost:11434')
|
|
18
22
|
self.base_url = base_url.rstrip('/')
|
|
19
23
|
self.model = model
|
|
20
24
|
self.timeout = timeout
|
|
@@ -4,13 +4,17 @@ Provides faster inference using vLLM OpenAI-compatible API
|
|
|
4
4
|
"""
|
|
5
5
|
|
|
6
6
|
import requests
|
|
7
|
+
import os
|
|
7
8
|
from typing import Iterator, List, Dict, Optional
|
|
8
9
|
|
|
9
10
|
|
|
10
11
|
class VLLMClient:
|
|
11
12
|
"""Client for vLLM server using OpenAI-compatible API"""
|
|
12
13
|
|
|
13
|
-
def __init__(self, base_url: str =
|
|
14
|
+
def __init__(self, base_url: str = None, model: str = "deepseek-coder-7b"):
|
|
15
|
+
# Use environment variable if available, otherwise default to localhost
|
|
16
|
+
if base_url is None:
|
|
17
|
+
base_url = os.getenv('VLLM_BASE_URL', 'http://localhost:8000')
|
|
14
18
|
self.base_url = base_url.rstrip('/')
|
|
15
19
|
self.model = model
|
|
16
20
|
self.api_url = f"{self.base_url}/v1"
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|