euriai 0.3.28__tar.gz → 0.3.30__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.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: euriai
3
- Version: 0.3.28
3
+ Version: 0.3.30
4
4
  Summary: Python client for EURI LLM API (euron.one) with CLI, LangChain, and LlamaIndex integration
5
5
  Author: euron.one
6
6
  Author-email: sudhanshu@euron.one
@@ -28,12 +28,12 @@ class EuriaiLlamaIndexLLM(LLM):
28
28
  temperature (float, optional): Sampling temperature. Defaults to 0.7.
29
29
  max_tokens (int, optional): Maximum number of tokens. Defaults to 1000.
30
30
  """
31
- # Create a dictionary of parameters for the parent class
31
+ # Create a dictionary of parameters with default values directly
32
32
  model_params = {
33
33
  "api_key": api_key,
34
- "model": model if model is not None else self.model,
35
- "temperature": temperature if temperature is not None else self.temperature,
36
- "max_tokens": max_tokens if max_tokens is not None else self.max_tokens,
34
+ "model": model if model is not None else "gpt-4.1-nano",
35
+ "temperature": temperature if temperature is not None else 0.7,
36
+ "max_tokens": max_tokens if max_tokens is not None else 1000,
37
37
  }
38
38
 
39
39
  # Initialize the parent class with the parameters
@@ -1,6 +1,6 @@
1
1
  import requests
2
2
  import numpy as np
3
- from typing import List, Optional
3
+ from typing import List, Optional, Callable, Any
4
4
  from llama_index.core.embeddings import BaseEmbedding
5
5
 
6
6
  class EuriaiLlamaIndexEmbedding(BaseEmbedding):
@@ -11,10 +11,10 @@ class EuriaiLlamaIndexEmbedding(BaseEmbedding):
11
11
 
12
12
  def __init__(self, api_key: str, model: Optional[str] = None):
13
13
  """Initialize embedding model with API key and model name."""
14
- # Create parameters for the parent class
14
+ # Create parameters for the parent class with default values directly
15
15
  embed_params = {
16
16
  "api_key": api_key,
17
- "model": model if model is not None else self.model,
17
+ "model": model if model is not None else "text-embedding-3-small",
18
18
  }
19
19
 
20
20
  # Initialize the parent class
@@ -38,6 +38,15 @@ class EuriaiLlamaIndexEmbedding(BaseEmbedding):
38
38
  """Get embedding for a single text."""
39
39
  return self._post_embedding([text])[0]
40
40
 
41
+ def get_query_embedding(self, query: str) -> List[float]:
42
+ """Get embedding for a query string."""
43
+ return self.get_text_embedding(query)
44
+
41
45
  def get_text_embeddings(self, texts: List[str]) -> List[List[float]]:
42
46
  """Get embeddings for multiple texts."""
43
- return self._post_embedding(texts)
47
+ return self._post_embedding(texts)
48
+
49
+ async def aget_query_embedding(self, query: str) -> List[float]:
50
+ """Async version of get_query_embedding."""
51
+ # We don't support async, so raise NotImplementedError
52
+ raise NotImplementedError("Async embeddings not supported")
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: euriai
3
- Version: 0.3.28
3
+ Version: 0.3.30
4
4
  Summary: Python client for EURI LLM API (euron.one) with CLI, LangChain, and LlamaIndex integration
5
5
  Author: euron.one
6
6
  Author-email: sudhanshu@euron.one
@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
2
2
 
3
3
  setup(
4
4
  name="euriai",
5
- version="0.3.28",
5
+ version="0.3.30",
6
6
  description="Python client for EURI LLM API (euron.one) with CLI, LangChain, and LlamaIndex integration",
7
7
  long_description=open("README.md", encoding="utf-8").read(),
8
8
  long_description_content_type="text/markdown",
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes