langroid 0.2.7__py3-none-any.whl → 0.2.9__py3-none-any.whl

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.
Files changed (7) hide show
  1. langroid/language_models/__init__.py +2 -0
  2. langroid/language_models/openai_gpt.py +17 -0
  3. {langroid-0.2.7.dist-info → langroid-0.2.9.dist-info}/METADATA +1 -1
  4. {langroid-0.2.7.dist-info → langroid-0.2.9.dist-info}/RECORD +7 -7
  5. pyproject.toml +1 -1
  6. {langroid-0.2.7.dist-info → langroid-0.2.9.dist-info}/LICENSE +0 -0
  7. {langroid-0.2.7.dist-info → langroid-0.2.9.dist-info}/WHEEL +0 -0
langroid/language_models/__init__.py CHANGED
@@ -16,6 +16,7 @@ from .base import (
16
16
  )
17
17
  from .openai_gpt import (
18
18
  OpenAIChatModel,
19
+ AnthropicModel,
19
20
  OpenAICompletionModel,
20
21
  OpenAIGPTConfig,
21
22
  OpenAIGPT,
@@ -39,6 +40,7 @@ __all__ = [
39
40
  "LLMTokenUsage",
40
41
  "LLMResponse",
41
42
  "OpenAIChatModel",
43
+ "AnthropicModel",
42
44
  "OpenAICompletionModel",
43
45
  "OpenAIGPTConfig",
44
46
  "OpenAIGPT",
langroid/language_models/openai_gpt.py CHANGED
@@ -65,6 +65,15 @@ OLLAMA_API_KEY = "ollama"
65
65
  DUMMY_API_KEY = "xxx"
66
66
 
67
67
 
68
+ class AnthropicModel(str, Enum):
69
+ """Enum for Anthropic models"""
70
+
71
+ CLAUDE_3_5_SONNET = "claude-3-5-sonnet-20240620"
72
+ CLAUDE_3_OPUS = "claude-3-opus-20240229"
73
+ CLAUDE_3_SONNET = "claude-3-sonnet-20240229"
74
+ CLAUDE_3_HAIKU = "claude-3-turbo-20240307"
75
+
76
+
68
77
  class OpenAIChatModel(str, Enum):
69
78
  """Enum for OpenAI Chat models"""
70
79
 
@@ -90,6 +99,10 @@ _context_length: Dict[str, int] = {
90
99
  OpenAIChatModel.GPT4_TURBO: 128_000,
91
100
  OpenAIChatModel.GPT4o: 128_000,
92
101
  OpenAICompletionModel.TEXT_DA_VINCI_003: 4096,
102
+ AnthropicModel.CLAUDE_3_5_SONNET: 200_000,
103
+ AnthropicModel.CLAUDE_3_OPUS: 200_000,
104
+ AnthropicModel.CLAUDE_3_SONNET: 200_000,
105
+ AnthropicModel.CLAUDE_3_HAIKU: 200_000,
93
106
  }
94
107
 
95
108
  _cost_per_1k_tokens: Dict[str, Tuple[float, float]] = {
@@ -99,6 +112,10 @@ _cost_per_1k_tokens: Dict[str, Tuple[float, float]] = {
99
112
  OpenAIChatModel.GPT4: (0.03, 0.06), # 8K context
100
113
  OpenAIChatModel.GPT4_TURBO: (0.01, 0.03), # 128K context
101
114
  OpenAIChatModel.GPT4o: (0.005, 0.015), # 128K context
115
+ AnthropicModel.CLAUDE_3_5_SONNET: (0.003, 0.015),
116
+ AnthropicModel.CLAUDE_3_OPUS: (0.015, 0.075),
117
+ AnthropicModel.CLAUDE_3_SONNET: (0.003, 0.015),
118
+ AnthropicModel.CLAUDE_3_HAIKU: (0.00025, 0.00125),
102
119
  }
103
120
 
104
121
 
{langroid-0.2.7.dist-info → langroid-0.2.9.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: langroid
3
- Version: 0.2.7
3
+ Version: 0.2.9
4
4
  Summary: Harness LLMs with Multi-Agent Programming
5
5
  License: MIT
6
6
  Author: Prasad Chalasani
{langroid-0.2.7.dist-info → langroid-0.2.9.dist-info}/RECORD RENAMED
@@ -62,12 +62,12 @@ langroid/embedding_models/protoc/embeddings_pb2.pyi,sha256=UkNy7BrNsmQm0vLb3NtGX
62
62
  langroid/embedding_models/protoc/embeddings_pb2_grpc.py,sha256=9dYQqkW3JPyBpSEjeGXTNpSqAkC-6FPtBHyteVob2Y8,2452
63
63
  langroid/embedding_models/remote_embeds.py,sha256=6_kjXByVbqhY9cGwl9R83ZcYC2km-nGieNNAo1McHaY,5151
64
64
  langroid/exceptions.py,sha256=w_Cr41nPAmsa6gW5nNFaO9yDcBCWdQqRspL1jYvZf5w,2209
65
- langroid/language_models/__init__.py,sha256=vrBtgR8Cq9UVfoI7nTms0IN7fd4y2JYpUP3GNV1DegY,898
65
+ langroid/language_models/__init__.py,sha256=1sUGobooTqq77XC7LxKsvME0RgSd5GGmeyrPo9SMh4U,940
66
66
  langroid/language_models/azure_openai.py,sha256=ncRCbKooqLVOY-PWQUIo9C3yTuKEFbAwyngXT_M4P7k,5989
67
67
  langroid/language_models/base.py,sha256=oAK2lXBqksMglqWqE2CjC03X3qPFXWgtjFWpH9hJ3C8,17500
68
68
  langroid/language_models/config.py,sha256=9Q8wk5a7RQr8LGMT_0WkpjY8S4ywK06SalVRjXlfCiI,378
69
69
  langroid/language_models/mock_lm.py,sha256=qdgj-wtbQBXlibo_0rIRfCt0hGTPRoxy1C4VjN6quI4,2707
70
- langroid/language_models/openai_gpt.py,sha256=U1B9glcJyYJYt1SWGQ8JPnzh2ntzms-wFl9y1ovcK4c,51018
70
+ langroid/language_models/openai_gpt.py,sha256=sALSYei56U57x8ugMayGIR6MtR9b9QsJ0MYhdt3_aMA,51674
71
71
  langroid/language_models/prompt_formatter/__init__.py,sha256=2-5cdE24XoFDhifOLl8yiscohil1ogbP1ECkYdBlBsk,372
72
72
  langroid/language_models/prompt_formatter/base.py,sha256=eDS1sgRNZVnoajwV_ZIha6cba5Dt8xjgzdRbPITwx3Q,1221
73
73
  langroid/language_models/prompt_formatter/hf_formatter.py,sha256=TFL6ppmeQWnzr6CKQzRZFYY810zE1mr8DZnhw6i85ok,5217
@@ -129,8 +129,8 @@ langroid/vector_store/meilisearch.py,sha256=6frB7GFWeWmeKzRfLZIvzRjllniZ1cYj3Hmh
129
129
  langroid/vector_store/momento.py,sha256=QaPzUnTwlswoawGB-paLtUPyLRvckFXLfLDfvbTzjNQ,10505
130
130
  langroid/vector_store/qdrant_cloud.py,sha256=3im4Mip0QXLkR6wiqVsjV1QvhSElfxdFSuDKddBDQ-4,188
131
131
  langroid/vector_store/qdrantdb.py,sha256=wYOuu5c2vIKn9ZgvTXcAiZXMpV8AOXEWFAzI8S8UP-0,16828
132
- pyproject.toml,sha256=sOrG9TDKiauO9bMWAotyGlQ8R-GQgLSt7ApBq1bhUKQ,6957
133
- langroid-0.2.7.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
134
- langroid-0.2.7.dist-info/METADATA,sha256=QqBF9VQobbivbIwaStn_2eMyVViqEx70iUtI5J37hBI,53950
135
- langroid-0.2.7.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
136
- langroid-0.2.7.dist-info/RECORD,,
132
+ pyproject.toml,sha256=vsVtb6-2IMl64e-3J3mqLEdL-hGAIw2X3gBjNk_HILs,6957
133
+ langroid-0.2.9.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
134
+ langroid-0.2.9.dist-info/METADATA,sha256=yColHB7bdfQ1nlYGucfcfH14xywCqm2IAjd8kPs3GhI,53950
135
+ langroid-0.2.9.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
136
+ langroid-0.2.9.dist-info/RECORD,,
pyproject.toml CHANGED
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "langroid"
3
- version = "0.2.7"
3
+ version = "0.2.9"
4
4
  description = "Harness LLMs with Multi-Agent Programming"
5
5
  authors = ["Prasad Chalasani <pchalasani@gmail.com>"]
6
6
  readme = "README.md"